Upload app.py

app.py

@@ -14,10 +14,15 @@ import streamlit as st
 from pathlib import Path
 from typing import Dict, Any, Optional
 
+# Add project root to path for agent imports
+project_root = Path(__file__).parent
+sys.path.insert(0, str(project_root))
+
 from src.full_pipeline.simple_pipeline import analyze_log_file
 
 from dotenv import load_dotenv
 from huggingface_hub import login as huggingface_login
+from huggingface_hub.utils import HfHubHTTPError
 
 load_dotenv()
 
@@ -109,11 +114,33 @@ def run_analysis(
         return {"success": False, "error": str(e)}
 
 
+@st.cache_resource
+def initialize_hf_login():
+    """Initialize Hugging Face login only once."""
+    hf_token = os.getenv("HF_TOKEN")
+    if hf_token:
+        try:
+            # Check if already logged in by trying to get user info
+            from huggingface_hub import whoami
+
+            whoami()
+            return True
+        except (HfHubHTTPError, Exception):
+            # Not logged in, try to login
+            try:
+                huggingface_login(token=hf_token)
+                return True
+            except Exception as e:
+                st.warning(f"Failed to login to Hugging Face: {e}")
+                return False
+    return False
+
+
 def main():
     """Main Streamlit app."""
 
-    if os.getenv("HF_TOKEN"):
-        huggingface_login(token=os.getenv("HF_TOKEN"))
+    # Initialize HF login (cached)
+    initialize_hf_login()
 
     st.set_page_config(
         page_title="Cybersecurity Agent Pipeline", page_icon="🛡️", layout="wide"