Update environment_setup

environment_setup

@@ -1,35 +1,52 @@
-# environment_setup.py
 import os
 import streamlit as st
 import gdown
 from pathlib import Path
 
 def setup_environment():
-    # Pfade definieren
-    BASE_PATH = Path(__file__).resolve().parent
-    VECTOR_STORE_PATH = BASE_PATH / "vector_stores"
-    VECTOR_STORE_PATH.mkdir(parents=True, exist_ok=True)
+    """Setup der Umgebung für die Anwendung"""
+    try:
+        # Pfade definieren
+        BASE_PATH = Path(__file__).resolve().parent
+        VECTOR_STORE_PATH = BASE_PATH / "vector_stores"
+        VECTOR_STORE_PATH.mkdir(parents=True, exist_ok=True)
+        
+        # Google Drive File ID für vectorstore.pkl
+        VECTOR_STORE_FILE = VECTOR_STORE_PATH / "vectorstore.pkl"
+        GDRIVE_FILE_ID = "YOUR_GOOGLE_DRIVE_FILE_ID"  # Ersetzen Sie dies mit Ihrer File-ID
+        
+        # Prüfe ob wir in Hugging Face Spaces sind
+        is_hf_space = "SPACE_ID" in os.environ
+        
+        if is_hf_space and not VECTOR_STORE_FILE.exists():
+            try:
+                # Lade von Google Drive nur wenn nötig
+                gdown.download(
+                    f"https://drive.google.com/uc?id={GDRIVE_FILE_ID}",
+                    str(VECTOR_STORE_FILE),
+                    quiet=False
+                )
+            except Exception as e:
+                st.error(f"Fehler beim Laden der Vector Store Datei: {str(e)}")
+                raise
 
-    # Google Drive File ID für vectorstore.pkl
-    VECTOR_STORE_FILE_ID = "1EAzau2hzMTu-f4ncCXaD9933HTay2o_8"  # Ersetzen Sie dies mit Ihrer File-ID
-    VECTOR_STORE_FILE = VECTOR_STORE_PATH / "vectorstore.pkl"
+        # API Key aus Hugging Face Secrets oder lokaler Umgebung
+        if is_hf_space:
+            api_key = os.environ.get('GOOGLE_API_KEY')
+            if not api_key:
+                st.error("Google API Key nicht gefunden in HF Secrets!")
+                raise ValueError("API Key fehlt")
+        else:
+            # Lokale Entwicklung - nutze .env oder direkte Umgebungsvariable
+            from dotenv import load_dotenv
+            load_dotenv()
+            api_key = os.environ.get('GOOGLE_API_KEY')
+            if not api_key:
+                st.error("Google API Key nicht in Umgebungsvariablen gefunden!")
+                raise ValueError("API Key fehlt")
 
-    # Nur vectorstore.pkl von Google Drive laden, wenn sie noch nicht existiert
-    if not VECTOR_STORE_FILE.exists():
-        try:
-            gdown.download(
-                f"https://drive.google.com/uc?id={VECTOR_STORE_FILE_ID}",
-                str(VECTOR_STORE_FILE),
-                quiet=False
-            )
-        except Exception as e:
-            st.error(f"Fehler beim Laden der Vector Store Datei: {str(e)}")
-            raise
+        return api_key, VECTOR_STORE_PATH
 
-    # API Key aus Hugging Face Secrets
-    api_key = os.environ.get('GOOGLE_API_KEY')
-    if not api_key:
-        st.error("Google API Key nicht gefunden in HF Secrets!")
-        raise ValueError("API Key fehlt")
-    
-    return api_key, VECTOR_STORE_PATH
+    except Exception as e:
+        st.error(f"Fehler beim Setup der Umgebung: {str(e)}")
+        raise