Update app.py

app.py

@@ -18,20 +18,68 @@ search_tool = DuckDuckGoSearchTool()
 ## TOOL 2
 class WikipediaTool(Tool):
     name = "wikipedia_search"
-    description = "Search Wikipedia and extract summaries"
+    description = "Search Wikipedia for a given topic and return a concise summary."
     inputs = {"query": "string"}
     output_type = "string"
-    
-    def forward(self, query):
-        return some_wikipedia_search_logic(query)
-wiki_tool = WikipediaTool()
+
+    def forward(self, query: str) -> str:
+        try:
+            page = wikipedia.page(query)
+            return page.summary[:1000]  # truncate to first 1000 characters
+        except wikipedia.exceptions.DisambiguationError as e:
+            return f"Disambiguation: {', '.join(e.options[:5])}"
+        except wikipedia.exceptions.PageError:
+            return "No Wikipedia page found."
+        except Exception as e:
+            return f"Error: {str(e)}"
+
+## TOOL 3
+class AudioTranscriptionTool(Tool):
+    name = "audio_transcriber"
+    description = "Transcribes an uploaded audio file to text using Whisper."
+    inputs = {"audio": "audio"}
+    output_type = "string"
+
+    def forward(self, audio: str) -> str:
+        try:
+            model = whisper.load_model("base")
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
+                tmp_file.write(audio.read())
+                tmp_file_path = tmp_file.name
+
+            result = model.transcribe(tmp_file_path)
+            os.remove(tmp_file_path)
+            return result["text"]
+        except Exception as e:
+            return f"Transcription failed: {str(e)}"
+
+## TOOL 4
+class ExcelReaderTool(Tool):
+    name = "excel_reader"
+    description = "Reads uploaded Excel file and summarizes sheet data."
+    inputs = {"excel": "file"}
+    output_type = "string"
+
+    def forward(self, excel) -> str:
+        try:
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx") as tmp_file:
+                tmp_file.write(excel.read())
+                tmp_file_path = tmp_file.name
+
+            df = pd.read_excel(tmp_file_path)
+            summary = df.describe(include="all").to_string()
+            os.remove(tmp_file_path)
+            return f"Summary statistics:\n{summary}"
+        except Exception as e:
+            return f"Failed to read Excel file: {str(e)}"
+
 
 async def run_and_submit_all(profile: gr.OAuthProfile | None):
     log_output = ""
 
     try:
         agent = ToolCallingAgent(
-            tools=[search_tool, wiki_tool],
+            tools=[search_tool, WikipediaTool(), AudioTranscriptionTool(), ExcelReaderTool()],
             model=InferenceClientModel(model="deepseek-ai/DeepSeek-V3", provider="together"),
             max_steps=15,
             verbosity_level=0,