Update Gradio_UI.py

Gradio_UI.py

@@ -17,41 +17,42 @@ class GradioUI:
             try:
                 print("Detected PDF upload and query. Calling document_qna_tool...")
 
-                # Dynamically find the tool by name
-                tool_name = "document_qna_tool"
+                # Dynamically find the tool function by actual function name
                 tool_func = None
                 for t in self.agent.tools:
-                    if hasattr(t, "name") and t.name == tool_name:
+                    # Some tools may not have the 'name' attribute; fall back to function __name__
+                    tool_name = getattr(t, "name", None) or getattr(t, "__name__", "")
+                    if tool_name == "document_qna_tool":
                         tool_func = t
                         break
 
                 if tool_func is None:
-                    return f"Error: Could not find tool named '{tool_name}'. Please check your app.py tool registration."
+                    return f"❌ Error: Could not find tool named 'document_qna_tool'. Please check your app.py tool registration."
 
                 # Call the tool with the uploaded PDF path and the user's question
                 response = tool_func(pdf_file.name, query)
-                print("Document Q&A tool finished.")
+                print("✅ Document Q&A tool finished.")
                 return response
 
             except Exception as e:
-                print(f"Error during Document Q&A tool execution: {e}")
+                print(f"❌ Error during Document Q&A tool execution: {e}")
                 return f"An error occurred during Document Q&A: {str(e)}"
 
         elif query and query.strip():
             print(f"No PDF file or query is for general task. Processing with agent.run(): {query}")
             try:
                 response = self.agent.run(query)
-                print("Agent.run finished for general query.")
+                print("✅ agent.run finished for general query.")
                 return response
             except Exception as e:
-                print(f"Error during agent.run: {e}")
+                print(f"❌ Error during agent.run: {e}")
                 return f"An error occurred while processing your request: {str(e)}"
 
         elif pdf_file is not None:
-            return "Please enter a question in the textbox to ask about the uploaded PDF."
+            return "⚠️ Please enter a question in the textbox to ask about the uploaded PDF."
 
         else:
-            return "Please enter a request or upload a document for analysis."
+            return "⚠️ Please enter a request or upload a document for analysis."
 
     def launch(self):
         """
@@ -93,4 +94,4 @@ class GradioUI:
                 outputs=agent_output
             )
 
-        demo.launch(share=False, inline=False)
+        demo.launch(share=False, inline=False)