Update app.py

app.py

@@ -62,6 +62,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
 embedding_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
 qa_pipeline = pipeline("text2text-generation", model="google/flan-t5-base")
 
+@Tool
 def document_qna_tool_fn(pdf_path: str, question: str) -> str:
     import traceback
 
@@ -97,14 +98,6 @@ def document_qna_tool_fn(pdf_path: str, question: str) -> str:
     except Exception as e:
         return f"[EXCEPTION] {type(e).__name__}: {str(e)}\n{traceback.format_exc()}"
 
-# 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()
 search_tool = DuckDuckGoSearchTool()
@@ -128,7 +121,7 @@ agent = CodeAgent(
         get_current_weather,
         image_generation_tool,
         search_tool,
-        document_qna_tool,  # Correctly wrapped tool
+        document_qna_tool,
         final_answer
     ],
     max_steps=6,