Update app.py

app.py

@@ -54,7 +54,7 @@ import nltk
 
 nltk.download('punkt')
 
-@tool
+@tool  # Register the function properly as a SmolAgents tool
 def fetch_latest_arxiv_papers(keywords: list, num_results: int = 5) -> list:
     """Fetches and ranks arXiv papers using BM25 keyword relevance.
 
@@ -68,8 +68,8 @@ def fetch_latest_arxiv_papers(keywords: list, num_results: int = 5) -> list:
     try:
         print(f"DEBUG: Searching arXiv papers with keywords: {keywords}")
 
-        # Broadly search keywords in title and abstract
-        query = "+OR+".join([f"(ti:\"{kw}\"+OR+abs:\"{kw}\")" for kw in keywords])  
+        # Use a general keyword search (without `ti:` and `abs:`)
+        query = "+AND+".join([f"all:{kw}" for kw in keywords])  
         query_encoded = urllib.parse.quote(query)
         url = f"http://export.arxiv.org/api/query?search_query={query_encoded}&start=0&max_results=50&sortBy=submittedDate&sortOrder=descending"
 
@@ -109,7 +109,6 @@ def fetch_latest_arxiv_papers(keywords: list, num_results: int = 5) -> list:
         return [{"error": f"Error fetching research papers: {str(e)}"}]
 
 
-
 # AI Model
 model = HfApiModel(
     max_tokens=2096,
@@ -135,7 +134,35 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Define Gradio Search Function
+# # Define Gradio Search Function
+# 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
+    
+#     if isinstance(results, list) and results and isinstance(results[0], dict):
+#         #Format output with better readability and clarity
+#         formatted_results = "\n\n".join([
+#             f"---\n\n"
+#             f"📌 **Title:**\n{paper['title']}\n\n"
+#             f"👨‍🔬 **Authors:**\n{paper['authors']}\n\n"
+#             f"📅 **Year:** {paper['year']}\n\n"
+#             f"📖 **Abstract:**\n{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()]  # Ensure valid keywords
     print(f"DEBUG: Received input keywords - {keywords}")  # Debug user input
@@ -146,25 +173,28 @@ def search_papers(user_input):
     
     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):
-        #Format output with better readability and clarity
         formatted_results = "\n\n".join([
             f"---\n\n"
-            f"📌 **Title:**\n{paper['title']}\n\n"
-            f"👨‍🔬 **Authors:**\n{paper['authors']}\n\n"
+            f"📌 **Title:** {paper['title']}\n\n"
+            f"👨‍🔬 **Authors:** {paper['authors']}\n\n"
             f"📅 **Year:** {paper['year']}\n\n"
-            f"📖 **Abstract:**\n{paper['abstract'][:500]}... *(truncated for readability)*\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."
 
 
-
 # Create Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# ScholarAgent")