Update app.py

app.py

@@ -1,4 +1,5 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool
+from smolagents.tools import Tool
 import datetime
 import requests
 import pytz
@@ -14,17 +15,7 @@ from transformers import pipeline
 API_KEY = os.getenv("Weather_Token")
 
 # -------------------- TOOL 1: Get Weather --------------------
-@tool
 def get_current_weather(place: str) -> str:
-    """
-    A tool that fetches the current weather of a particular place.
-
-    Args:
-        place (str): A string representing a valid place (e.g., 'London/Paris').
-
-    Returns:
-        str: Weather description including condition, temperature, humidity, and wind speed.
-    """
     api_key = API_KEY
     url = "https://api.openweathermap.org/data/2.5/weather"
     params = {
@@ -32,17 +23,14 @@ def get_current_weather(place: str) -> str:
         "appid": api_key,
         "units": "metric"
     }
-
     try:
         response = requests.get(url, params=params)
         data = response.json()
-
         if response.status_code == 200:
             weather_desc = data["weather"][0]["description"]
             temperature = data["main"]["temp"]
             humidity = data["main"]["humidity"]
             wind_speed = data["wind"]["speed"]
-
             return (
                 f"Weather in {place}:\n"
                 f"- Condition: {weather_desc}\n"
@@ -55,19 +43,10 @@ def get_current_weather(place: str) -> str:
     except Exception as e:
         return f"Error fetching weather data for '{place}': {str(e)}"
 
+get_current_weather = Tool(name="get_current_weather", description="Fetch current weather for a given place")(get_current_weather)
 
 # -------------------- TOOL 2: Get Time --------------------
-@tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """
-    A tool that fetches the current local time in a specified timezone.
-
-    Args:
-        timezone (str): A string representing a valid timezone (e.g., 'America/New_York').
-
-    Returns:
-        str: The current local time formatted as a string.
-    """
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -75,30 +54,14 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+get_current_time_in_timezone = Tool(name="get_current_time_in_timezone", description="Fetch local time for a given timezone")(get_current_time_in_timezone)
 
 # -------------------- TOOL 3: Document QnA --------------------
 embedding_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
 qa_pipeline = pipeline("text2text-generation", model="google/flan-t5-base")
 
-from smolagents.tools import Tool
-
-@Tool(name="document_qna_tool", description="Answer questions about a PDF document.")
 def document_qna_tool(pdf_path: str, question: str) -> str:
-
-    """
-    A tool that answers natural language questions about a given PDF document.
-    
-    Args:
-        pdf_path (str): Path to the local PDF file.
-        question (str): Question about the content of the PDF.
-    
-    Returns:
-        str: Answer to the question based on the content.
-    """
     import os, fitz, traceback
-    from sentence_transformers import SentenceTransformer, util
-    from transformers import pipeline
-
     try:
         print(f"[DEBUG] PDF Path: {pdf_path}")
         print(f"[DEBUG] Question: {question}")
@@ -106,50 +69,32 @@ def document_qna_tool(pdf_path: str, question: str) -> str:
         if not os.path.exists(pdf_path):
             return f"[ERROR] File not found: {pdf_path}"
 
-        print("[DEBUG] Opening PDF...")
-        try:
-            doc = fitz.open(pdf_path)
-        except RuntimeError as e:
-            return f"[ERROR] Could not open PDF. It may be corrupted or encrypted. Details: {str(e)}"
-
-        text_chunks = []
-        for page in doc:
-            text = page.get_text()
-            if text.strip():
-                text_chunks.append(text)
+        doc = fitz.open(pdf_path)
+        text_chunks = [page.get_text() for page in doc if page.get_text().strip()]
         doc.close()
 
         if not text_chunks:
             return "[ERROR] No readable text in the PDF."
 
-        print(f"[DEBUG] Extracted {len(text_chunks)} text chunks.")
-        print(f"[DEBUG] First text chunk preview:\n{text_chunks[0][:300]}...")
-
         embeddings = embedding_model.encode(text_chunks, convert_to_tensor=True)
         question_embedding = embedding_model.encode(question, convert_to_tensor=True)
 
-        print("[DEBUG] Performing semantic search...")
         scores = util.pytorch_cos_sim(question_embedding, embeddings)[0]
-
-        print(f"[DEBUG] Similarity scores: {scores}")
-
         if scores.shape[0] == 0:
             return "[ERROR] No semantic matches found in PDF text."
 
         best_match_idx = scores.argmax().item()
         best_context = text_chunks[best_match_idx]
 
-        print(f"[DEBUG] Best context preview:\n{best_context[:300]}...")
-
         prompt = f"Context: {best_context}\nQuestion: {question}"
-        print("[DEBUG] Calling QA model...")
         answer = qa_pipeline(prompt, max_new_tokens=100)[0]['generated_text']
-
         return f"Answer: {answer.strip()}"
 
     except Exception as e:
         return f"[EXCEPTION] {type(e).__name__}: {str(e)}\n{traceback.format_exc()}"
 
+document_qna_tool = Tool(name="document_qna_tool", description="Answer questions about a PDF document")(document_qna_tool)
+
 # -------------------- Other Components --------------------
 final_answer = FinalAnswerTool()
 search_tool = DuckDuckGoSearchTool()
@@ -173,7 +118,7 @@ agent = CodeAgent(
         get_current_weather,
         image_generation_tool,
         search_tool,
-        document_qna_tool,  # ← New Tool Added
+        document_qna_tool,
         final_answer
     ],
     max_steps=6,
@@ -184,8 +129,9 @@ agent = CodeAgent(
     description=None,
     prompt_templates=prompt_templates
 )
+
 print("[DEBUG] Registered Tools:")
 for t in agent.tools:
     print(f" - {getattr(t, 'name', str(t))}")
 
-GradioUI(agent).launch()
+GradioUI(agent).launch()