Update app.py

app.py

@@ -1,16 +1,17 @@
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
-import yaml
 import feedparser
+import urllib.parse
+import yaml
 import gradio as gr
+from smolagents import CodeAgent, HfApiModel, tool
 
-import urllib.parse  # Import URL encoder
 
-def fetch_latest_arxiv_papers(keywords: list, num_results: int = 1) -> list:
+@tool
+def fetch_latest_arxiv_papers(keywords: list, num_results: int = 3) -> list:
     """Fetches the latest research papers from arXiv based on provided keywords."""
     try:
         print(f"DEBUG: Searching arXiv papers with keywords: {keywords}")  # Debug input
         
-        # Properly format query with +AND+ for multiple keywords
+        # ✅ Properly format query with +AND+ for multiple keywords
         query = "+AND+".join([f"all:{kw}" for kw in keywords])  
         query_encoded = urllib.parse.quote(query)  # Encode spaces and special characters
         
@@ -36,8 +37,6 @@ def fetch_latest_arxiv_papers(keywords: list, num_results: int = 1) -> list:
         print(f"ERROR: {str(e)}")  # Debug errors
         return [f"Error fetching research papers: {str(e)}"]
 
-
-
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -45,12 +44,14 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
+# Load prompt templates
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
+# Create the AI Agent
 agent = CodeAgent(
     model=model,
-    tools=[fetch_latest_arxiv_papers],
+    tools=[fetch_latest_arxiv_papers],  # Properly registered tool
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -60,6 +61,7 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
+# 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
@@ -80,8 +82,7 @@ def search_papers(user_input):
     print("DEBUG: No results found.")
     return "No results found. Try different keywords."
 
-
-# Create a simple Gradio interface
+#  Create Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# arXiv Research Paper Fetcher")
     keyword_input = gr.Textbox(label="Enter keywords (comma-separated)", placeholder="e.g., deep learning, reinforcement learning")
@@ -92,4 +93,5 @@ with gr.Blocks() as demo:
 
     print("DEBUG: Gradio UI is running. Waiting for user input...")
 
+# Launch Gradio App
 demo.launch()