Update app.py

app.py

@@ -256,75 +256,38 @@ agent = CodeAgent(
 
 
 
-# #Search Papers
-# 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
-    
-#     if not keywords:
-#         print("DEBUG: No valid keywords provided.")
-#         return "Error: Please enter at least one valid keyword."
-    
-#     results = fetch_latest_arxiv_papers(keywords, num_results=3)  # Fetch 3 results
-#     print(f"DEBUG: Results received - {results}")  # Debug function output
-
-#     # Check if the API returned an error
-#     if isinstance(results, list) and len(results) > 0 and "error" in results[0]:
-#         return results[0]["error"]  # Return the error message directly
-
-#     # Format results only if valid papers exist
-#     if isinstance(results, list) and results and isinstance(results[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 results
-#         ])
-#         return formatted_results
-
-#     print("DEBUG: No results found.")
-#     return "No results found. Try different keywords."
-
+#Search Papers
 def search_papers(user_input):
-    keywords = [kw.strip() for kw in user_input.split(",") if kw.strip()]
-    print(f"DEBUG: Received input keywords - {keywords}")
-
+    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
+    
     if not keywords:
+        print("DEBUG: No valid keywords provided.")
         return "Error: Please enter at least one valid keyword."
+    
+    results = fetch_latest_arxiv_papers(keywords, num_results=3)  # Fetch 3 results
+    print(f"DEBUG: Results received - {results}")  # Debug function output
+
+    # Check if the API returned an error
+    if isinstance(results, list) and len(results) > 0 and "error" in results[0]:
+        return results[0]["error"]  # Return the error message directly
+
+    # Format results only if valid papers exist
+    if isinstance(results, list) and results and isinstance(results[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 results
+        ])
+        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
-# GradioUI(agent).launch()
 
 
 # Create Gradio UI