Update app.py

app.py

@@ -289,36 +289,38 @@ agent = CodeAgent(
 #     return "No results found. Try different keywords."
 
 def search_papers(user_input):
-    keywords = [kw.strip() for kw in user_input.split(",") if kw.strip()]  # Ensure valid keywords
-    print(f"DEBUG: Received input keywords - {keywords}")  # Debug user input
-    
+    keywords = [kw.strip() for kw in user_input.split(",") if kw.strip()]
+    print(f"DEBUG: Received input keywords - {keywords}")
+
     if not keywords:
-        print("DEBUG: No valid keywords provided.")
         return "Error: Please enter at least one valid keyword."
-    
-    # Use CodeAgent to process the request
-    response = agent.run(user_input)  # Now it actually uses CodeAgent
-    print(f"DEBUG: Agent Response - {response}")  # Debug response from the agent
-
-    # Check if response is valid
-    if isinstance(response, list) and len(response) > 0 and "error" in response[0]:
-        return response[0]["error"]  # Return the error message directly
-
-    # Format results only if valid papers exist
-    if isinstance(response, list) and response and isinstance(response[0], dict):
-        formatted_results = "\n\n".join([
-            f"---\n\n"
-            f"📌 **Title:** {paper['title']}\n\n"
-            f"👨‍🔬 **Authors:** {paper['authors']}\n\n"
-            f"📅 **Year:** {paper['year']}\n\n"
-            f"📖 **Abstract:** {paper['abstract'][:500]}... *(truncated for readability)*\n\n"
-            f"[🔗 Read Full Paper]({paper['link']})\n\n"
-            for paper in response
-        ])
-        return formatted_results
-
-    print("DEBUG: No results found.")
-    return "No results found. Try different keywords."
+
+    # Check if 'final_answer' exists in prompt_templates
+    if "final_answer" not in agent.prompt_templates:
+        agent.prompt_templates["final_answer"] = {
+            "pre_messages": "Here is the final answer based on the research:",
+            "post_messages": "End of response."
+        }
+
+    try:
+        response = agent.run(user_input)  # Now it actually uses CodeAgent
+        print(f"DEBUG: Agent Response - {response}")
+
+        if isinstance(response, list) and response and isinstance(response[0], dict):
+            formatted_results = "\n\n".join([
+                f"📌 **Title:** {paper['title']}\n\n"
+                f"👨‍🔬 **Authors:** {paper['authors']}\n\n"
+                f"📅 **Year:** {paper['year']}\n\n"
+                f"📖 **Abstract:** {paper['abstract'][:500]}... *(truncated for readability)*\n\n"
+                f"[🔗 Read Full Paper]({paper['link']})\n\n"
+                for paper in response
+            ])
+            return formatted_results
+
+        return "No results found. Try different keywords."
+    except Exception as e:
+        print(f"ERROR: {e}")
+        return f"An error occurred: {str(e)}"
 
 
 # # Launch Gradio UI with CodeAgent