Update app/main.py

app/main.py

@@ -5,10 +5,22 @@ from fastapi.responses import StreamingResponse, JSONResponse
 from pydantic import BaseModel
 from gradio_client import Client
 import httpx
+import time
+import json
 
-# Initialize Gradio client
+# Load environment variables
+load_dotenv()
+
+# Configure from environment
+SPACE_ID = os.getenv("SPACE_ID", "prithivMLmods/SAMBANOVA")
+DEFAULT_API = os.getenv("DEFAULT_API", "/chat")
+GRADIO_TIMEOUT = int(os.getenv("GRADIO_TIMEOUT", "60"))
+API_KEY = os.getenv("API_KEY")
+if not API_KEY:
+    raise RuntimeError("Missing API_KEY in environment")
+
+# Initialize Gradio client and set timeout
 client = Client(SPACE_ID)
-# Override the default HTTPX timeout to avoid read timeouts
 client.client.timeout = httpx.Timeout(timeout=GRADIO_TIMEOUT)
 
 
@@ -19,7 +31,6 @@ def chat_with_gradio(message: str, api_name: str = DEFAULT_API):
     try:
         return client.predict(message=message, api_name=api_name)
     except Exception as e:
-        # Specific handling for read timeout
         msg = str(e)
         if "ReadTimeout" in msg:
             raise RuntimeError(f"Gradio API timed out after {GRADIO_TIMEOUT}s")