test2

app.py

@@ -1,8 +1,9 @@
-from openai import OpenAI
 import os
-import math
 import datetime
 import requests
+import gradio as gr
+import pandas as pd
+from openai import OpenAI
 
 class ToolEnhancedAgent:
     def __init__(self):
@@ -10,7 +11,7 @@ class ToolEnhancedAgent:
         if not api_key:
             raise ValueError("OPENAI_API_KEY is not set.")
         self.client = OpenAI(api_key=api_key)
-        print("ToolEnhancedAgent initialized with GPT + CoT + Tools.")
+        print("✅ ToolEnhancedAgent initialized with GPT-4 + CoT + Tool usage.")
 
     def use_tool(self, tool_name: str, input_text: str) -> str:
         try:
@@ -19,26 +20,25 @@ class ToolEnhancedAgent:
             elif tool_name == "date":
                 return str(datetime.datetime.now().date())
             elif tool_name == "wikipedia":
-                summary = self.search_wikipedia(input_text)
-                return summary
+                return self.search_wikipedia(input_text)
             else:
                 return "[Tool Error: Unknown Tool]"
         except Exception as e:
             return f"[Tool Error: {e}]"
 
-    def search_wikipedia(self, query):
+    def search_wikipedia(self, query: str) -> str:
         try:
             res = requests.get(f"https://en.wikipedia.org/api/rest_v1/page/summary/{query}")
             if res.status_code == 200:
                 return res.json().get("extract", "No summary found.")
             else:
-                return f"No Wikipedia summary for {query}."
+                return f"No Wikipedia summary found for '{query}'."
         except Exception as e:
             return f"Wiki error: {e}"
 
     def __call__(self, question: str) -> str:
         prompt = (
-            "You are a helpful AI assistant. Use tools when necessary. "
+            "You are a helpful AI assistant. Use tools when needed. "
             "Think step-by-step before answering. Respond clearly.\n\n"
             f"Question: {question}\n"
             "Answer (show thinking steps):"
@@ -55,7 +55,7 @@ class ToolEnhancedAgent:
                 max_tokens=700,
             )
             answer = response.choices[0].message.content.strip()
-            print(f"Answer generated: {answer[:100]}...")
+            print(f"📤 Agent response: {answer[:100]}...")
             return answer
         except Exception as e:
             print(f"[Agent Error]: {e}")