Upload folder using huggingface_hub

Dockerfile

@@ -1,12 +1,32 @@
 FROM python:3.10-slim
 
+# Install system dependencies including DNS tools
+RUN apt-get update && apt-get install -y \
+    curl \
+    dnsutils \
+    && rm -rf /var/lib/apt/lists/*
+
 WORKDIR /code
 
+# Set DNS configuration
+RUN echo "nameserver 8.8.8.8" > /etc/resolv.conf && \
+    echo "nameserver 8.8.4.4" >> /etc/resolv.conf
+
+# Set network timeouts
+ENV CURL_TIMEOUT=30
+ENV REQUESTS_TIMEOUT=30
+
+# Copy and install requirements
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Copy application code
 COPY . .
 
+# Add healthcheck
+HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:7860/ || exit 1
+
 EXPOSE 7860
 
 CMD ["python", "app.py"]

app.py

@@ -11,6 +11,10 @@ from typing import Dict, Any, List
 import json
 from datetime import datetime
 import logging
+import os
+import socket
+import requests
+from requests.adapters import HTTPAdapter, Retry
 
 from agentic_system import AgenticSystem
 from team_management import TeamManager
@@ -21,11 +25,41 @@ from reasoning.unified_engine import UnifiedReasoningEngine
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Configure network settings
+TIMEOUT = int(os.getenv('REQUESTS_TIMEOUT', '30'))
+RETRIES = 3
+
+# Configure session with retries
+session = requests.Session()
+retries = Retry(total=RETRIES,
+                backoff_factor=1,
+                status_forcelist=[429, 500, 502, 503, 504])
+session.mount('https://', HTTPAdapter(max_retries=retries))
+session.mount('http://', HTTPAdapter(max_retries=retries))
+
+def check_network():
+    """Check network connectivity."""
+    try:
+        # Try to resolve huggingface.co
+        socket.gethostbyname('huggingface.co')
+        # Try to connect to the API
+        response = session.get('https://huggingface.co/api/health', 
+                             timeout=TIMEOUT)
+        response.raise_for_status()
+        return True
+    except Exception as e:
+        logger.error(f"Network check failed: {e}")
+        return False
+
 class AgentInterface:
     """Interface for the agentic system."""
     
     def __init__(self):
         """Initialize the interface components."""
+        # Check network connectivity
+        if not check_network():
+            logger.warning("Network connectivity issues detected")
+            
         self.orchestrator = AgentOrchestrator()
         self.reasoning_engine = UnifiedReasoningEngine(
             min_confidence=0.7,
@@ -49,61 +83,49 @@ class AgentInterface:
                 context=context
             )
             
-            if result.success:
-                return result.answer
-            else:
-                return f"Error: Unable to process query. Please try again."
-                
+            return result.get('answer', 'No response generated')
+            
         except Exception as e:
             logger.error(f"Error processing query: {e}")
-            return f"Error: {str(e)}"
+            return f"An error occurred: {str(e)}"
 
 # Initialize interface
 interface = AgentInterface()
 
 # Create Gradio interface
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("""
-    # AI Reasoning System
-    This system uses advanced reasoning strategies including local LLM for improved performance.
-    
-    Note: First query might take a few seconds as the model loads.
-    """)
+with gr.Blocks(title="Advanced Reasoning System") as demo:
+    gr.Markdown("# Advanced Reasoning System")
     
     with gr.Row():
-        with gr.Column(scale=4):
+        with gr.Column():
             input_text = gr.Textbox(
-                label="Your question",
-                placeholder="Ask me anything...",
-                lines=2
+                label="Your Query",
+                placeholder="Enter your query here...",
+                lines=3
             )
+            submit_btn = gr.Button("Submit")
+        
+        with gr.Column():
             output_text = gr.Textbox(
                 label="Response",
-                lines=10,
-                interactive=False
+                lines=5
             )
-            submit_btn = gr.Button("Ask")
-            clear_btn = gr.Button("Clear")
-            
-        with gr.Column(scale=1):
-            gr.Markdown("""
-            ### Example Questions:
-            - What are the implications of artificial intelligence on society?
-            - How does climate change affect global ecosystems?
-            - What are the philosophical implications of quantum mechanics?
-            """)
     
-    # Set up event handlers
     submit_btn.click(
         fn=interface.process_query,
         inputs=input_text,
         outputs=output_text
     )
-    clear_btn.click(
-        lambda: ("", ""),
-        inputs=None,
-        outputs=[input_text, output_text]
-    )
+
+# Start the server with health check endpoint
+app = gr.mount_gradio_app(None, demo, path="/")
+
+@app.get("/health")
+async def health_check():
+    """Health check endpoint."""
+    if check_network():
+        return {"status": "healthy"}
+    return {"status": "degraded", "reason": "network_issues"}
 
 # Launch the interface
 if __name__ == "__main__":