Update app.py

app.py

@@ -15,22 +15,25 @@ from transformers import pipeline
 API_KEY = os.getenv("Weather_Token")
 
 # -------------------- TOOL 1: Get Weather --------------------
+@Tool
 def get_current_weather(place: str) -> str:
-    api_key = API_KEY
-    url = "https://api.openweathermap.org/data/2.5/weather"
-    params = {
-        "q": place,
-        "appid": api_key,
-        "units": "metric"
-    }
     try:
+        url = "https://api.openweathermap.org/data/2.5/weather"
+        params = {
+            "q": place,
+            "appid": API_KEY,
+            "units": "metric"
+        }
+
         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"
@@ -43,9 +46,9 @@ 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:
     try:
         tz = pytz.timezone(timezone)
@@ -54,14 +57,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")
 
-def document_qna_tool(pdf_path: str, question: str) -> str:
-    import os, fitz, traceback
+def document_qna_tool_fn(pdf_path: str, question: str) -> str:
+    import traceback
+
     try:
         print(f"[DEBUG] PDF Path: {pdf_path}")
         print(f"[DEBUG] Question: {question}")
@@ -88,12 +91,19 @@ def document_qna_tool(pdf_path: str, question: str) -> str:
 
         prompt = f"Context: {best_context}\nQuestion: {question}"
         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)
+# Wrap the function using Tool(...)
+document_qna_tool = Tool(
+    name="document_qna_tool",
+    description="Answer questions about a PDF document.",
+    func=document_qna_tool_fn
+)
+
 
 # -------------------- Other Components --------------------
 final_answer = FinalAnswerTool()
@@ -118,7 +128,7 @@ agent = CodeAgent(
         get_current_weather,
         image_generation_tool,
         search_tool,
-        document_qna_tool,
+        document_qna_tool,  # Correctly wrapped tool
         final_answer
     ],
     max_steps=6,