Added networkx dependency

requirements.txt

@@ -1,7 +1,7 @@
 # Core dependencies
 fastapi>=0.68.0
 uvicorn>=0.15.0
-gradio==4.44.1
+gradio>=4.44.1
 pydantic>=2.0.0
 python-dotenv>=0.19.0
 
@@ -17,6 +17,7 @@ numpy>=1.24.0
 pandas>=2.1.0
 scikit-learn>=1.3.2
 plotly>=5.18.0
+networkx>=3.2.1  # Added for graph operations
 
 # Model integration
 huggingface-hub>=0.19.4

space.yml

@@ -20,8 +20,7 @@ compute:
 # Environment setup
 env:
   - MODEL_BACKEND=groq
-  - GROQ_API_KEY
-  - HUGGINGFACE_TOKEN
+  - GROQ_API_KEY  # This will be loaded from repository secrets
   - ENABLE_LOCAL_FALLBACK=true
   - CACHE_MODELS=false
   - GRADIO_SERVER_PORT=7860
@@ -47,9 +46,8 @@ models:
 # Dependencies
 dependencies:
   python: 
-    - "gradio==4.44.1"
+    - "gradio>=4.44.1"
     - "groq>=0.4.1"
-    - "huggingface-hub>=0.19.4"
     - "fastapi>=0.68.0"
     - "uvicorn>=0.15.0"
     - "pydantic>=2.0.0"
@@ -60,6 +58,7 @@ dependencies:
     - "pandas>=2.1.0"
     - "scikit-learn>=1.3.2"
     - "plotly>=5.18.0"
+    - "networkx>=3.2.1"  # Added for graph operations
   system:
     - git-lfs
     - cmake

startup.sh

@@ -5,6 +5,12 @@ set -e
 
 echo "Starting initialization..."
 
+# Check for required environment variables
+if [ -z "$GROQ_API_KEY" ]; then
+    echo "Error: GROQ_API_KEY is not set. Please add it as a secret in your Hugging Face space settings."
+    exit 1
+fi
+
 # Update pip and install dependencies
 echo "Updating pip and installing dependencies..."
 python -m pip install --upgrade pip