Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import time
 import json
 import httpx
 import gradio as gr
@@ -55,9 +56,9 @@ def refine_response(user_input):
         deepseek_response = query_deepseek(user_input)
 
         # If either response failed, return the available one
-        if "Error" in gemma_response:
+        if not gemma_response.strip():
             return f"Only DeepSeek Response:\n{deepseek_response}"
-        if "Error" in deepseek_response:
+        if not deepseek_response.strip():
             return f"Only Gemma Response:\n{gemma_response}"
 
         # Prepare refinement prompt
@@ -69,27 +70,46 @@ def refine_response(user_input):
 
         Please combine the best elements of both, improve clarity, and provide a final refined answer.
         """
-        
-        # Send request to OpenRouter
-        response = httpx.post(
-            "https://openrouter.ai/api/v1/chat/completions",
-            headers={
-                "Authorization": f"Bearer {OPENROUTER_API_KEY}",
-                "Content-Type": "application/json"
-            },
-            json={
-                "model": "deepseek/deepseek-r1:free",
-                "messages": [{"role": "user", "content": improvement_prompt}]
-            }
-        )
 
-        # Print raw response for debugging
-        print("OpenRouter Response:", response.text)
+        # Retry loop for OpenRouter API
+        max_retries = 3
+        for attempt in range(max_retries):
+            try:
+                response = httpx.post(
+                    "https://openrouter.ai/api/v1/chat/completions",
+                    headers={
+                        "Authorization": f"Bearer {OPENROUTER_API_KEY}",
+                        "Content-Type": "application/json"
+                    },
+                    json={
+                        "model": "deepseek/deepseek-r1:free",
+                        "messages": [{"role": "user", "content": improvement_prompt}]
+                    },
+                    timeout=30  # Increase timeout to 30 seconds
+                )
+
+                # Debugging: Print response
+                print(f"Attempt {attempt + 1}: OpenRouter Response:", response.text)
+
+                # Check if the response is valid JSON
+                response_json = response.json()
+                
+                # Extract the refined response
+                refined_content = response_json["choices"][0]["message"]["content"]
+
+                # If DeepSeek gives an empty response, retry
+                if refined_content.strip():
+                    return refined_content
+                else:
+                    print("Received empty response from DeepSeek, retrying...")
+                    time.sleep(2)  # Wait before retrying
+
+            except Exception as e:
+                print(f"Error on attempt {attempt + 1}: {str(e)}")
+                time.sleep(2)  # Wait before retrying
 
-        # Check if response is valid JSON
-        response_json = response.json()
-        
-        return response_json["choices"][0]["message"]["content"]
+        # If all retries fail, return the better of the two initial responses
+        return f"Refinement failed. Here’s the best available response:\n\n{max(gemma_response, deepseek_response, key=len)}"
 
     except Exception as e:
         return f"Error refining response: {str(e)}"