Update text2generation.py

text2generation.py

@@ -3,23 +3,37 @@ import asyncio
 import httpx
 from dotenv import load_dotenv
 
-load_dotenv()
+load_dotenv()  # Load .env file if you use one
 
-async def generate_long_answer(question):
-    async with httpx.AsyncClient(timeout=10.0) as client:
-        response = await client.post(
-            "https://openrouter.ai/api/v1/chat/completions",
-            headers={
-                "Authorization": f"Bearer {os.getenv('OPENROUTER_API_KEY')}",
-                "Content-Type": "application/json",
-                "HTTP-Referer": "https://sreepathi-ravikumar-sample.hf.space",
-                "X-Title": "Educational AI Assistant"
-            },
-            json={
-                "model": "deepseek/deepseek-chat-v3-0324:free",
-                "messages": [{"role": "user", "content": question}],
-                "temperature": 0.3,
-                "max_tokens": 700
-            }
-        )
-        return response.json()['choices'][0]['message']['content']
+async def generate_long_answer(question: str) -> str:
+    headers = {
+        "Authorization": f"Bearer {os.getenv('OPENROUTER_API_KEY')}",
+        "Content-Type": "application/json",
+        "HTTP-Referer": "https://sreepathi-ravikumar-sample.hf.space",  # or your real referrer
+        "X-Title": "Educational AI Assistant"
+    }
+
+    payload = {
+        "model": "deepseek/deepseek-chat-v3-0324:free",
+        "messages": [{
+            "role": "user",
+            "content": f"Provide a comprehensive educational answer:\n\n{question}"
+        }],
+        "temperature": 0.3,
+        "max_tokens": 700
+    }
+
+    try:
+        async with httpx.AsyncClient(timeout=30.0, http2=True) as client:
+            response = await client.post(
+                "https://openrouter.ai/api/v1/chat/completions",
+                headers=headers,
+                json=payload
+            )
+            response.raise_for_status()
+            return response.json()["choices"][0]["message"]["content"]
+    
+    except httpx.HTTPStatusError as http_err:
+        return f"HTTP error: {http_err.response.status_code} - {http_err.response.text}"
+    except Exception as e:
+        return f"Error generating answer: {str(e)}"