Update app.py

app.py

@@ -207,6 +207,64 @@ agent = workflow.compile()
 
 # 6. Run Function
 
+def get_concise_answer(
+    prompt: str,
+    model: str = "deepseek/deepseek-chat-v3-0324:free",
+    timeout: int = 1000,
+) -> Optional[str]:
+    """
+    Fetches a concise AI response from OpenRouter API.
+    
+    Args:
+        prompt (str): The user's input/question.
+        model (str): The AI model to use (default: deepseek-chat-v3-0324).
+        timeout (int): Request timeout in seconds (default: 10).
+    
+    Returns:
+        str: The concise AI response, or None if an error occurs.
+    """
+    try:
+        response = requests.post(
+            url="https://openrouter.ai/api/v1/chat/completions",
+            headers=HEADERS,  # Ensure HEADERS is defined elsewhere
+            data=json.dumps({
+                "model": model,
+                "messages": [
+                    {
+                        "role": "system",
+                        "content": """
+                        You are an AI assistant that answers questions directly and concisely. 
+                        Respond with only the factual answer, no prefixes, explanations, or extra text. 
+                        Examples:
+                        - User: Who is the CEO of Tesla?
+                        - AI: Elon Musk
+                        - User: Capital of Japan?
+                        - AI: Tokyo
+                        """
+                    },
+                    {"role": "user", "content": prompt}
+                ],
+            }),
+            timeout=timeout,
+        )
+        response.raise_for_status()  # Raises HTTPError for bad responses (4xx, 5xx)
+        data = response.json()
+
+        # Check if response has the expected structure
+        if "choices" not in data or not data["choices"]:
+            return None
+        
+        model_response = data["choices"][0]["message"]["content"]
+        return model_response.strip()  # Remove extra whitespace
+
+    except requests.exceptions.RequestException as e:
+        print(f"API Request failed: {e}")
+        return None
+    except (KeyError, json.JSONDecodeError) as e:
+        print(f"Failed to parse API response: {e}")
+        return None
+
+'''
 def get_concise_answer(prompt):
   response = requests.post(
     url="https://openrouter.ai/api/v1/chat/completions",
@@ -241,7 +299,7 @@ def get_concise_answer(prompt):
   data = response.json()
   model_response = data['choices'][0]['message']['content']
   return model_response
-
+'''
 def run_groq_agent(query: str):
     response = agent.invoke({
         "messages": [HumanMessage(content=query)],