Update app.py

app.py

@@ -21,16 +21,12 @@ import wikipedia
 class WikipediaTool(Tool):
     name = "wikipedia_search"
     description = "Search Wikipedia for a given topic and return a concise summary."
-    
-    parameters = {
-        "type": "object",
-        "properties": {
-            "query": {
-                "type": "string",
-                "description": "The topic to search on Wikipedia"
-            }
-        },
-        "required": ["query"]
+
+    inputs = {
+        "query": {
+            "type": "string",
+            "description": "The topic to search on Wikipedia"
+        }
     }
 
     output_type = "string"
@@ -38,7 +34,7 @@ class WikipediaTool(Tool):
     def forward(self, query: str) -> str:
         try:
             page = wikipedia.page(query)
-            return page.summary[:1000]
+            return page.summary[:1000]  # First 1000 characters
         except wikipedia.exceptions.DisambiguationError as e:
             return f"Disambiguation: {', '.join(e.options[:5])}"
         except wikipedia.exceptions.PageError:
@@ -49,78 +45,70 @@ class WikipediaTool(Tool):
 
 
 # --- Tool 2: Audio Transcription Tool (uses whisper base) ---
-import base64
 import whisper
 import tempfile
-import os
+import base64
 
 class AudioTranscriptionTool(Tool):
     name = "audio_transcriber"
     description = "Transcribes an uploaded audio file to text using Whisper."
-    
-    parameters = {
-        "type": "object",
-        "properties": {
-            "audio": {
-                "type": "string",
-                "description": "The audio file in base64-encoded format"
-            }
-        },
-        "required": ["audio"]
+
+    inputs = {
+        "audio": {
+            "type": "string",  # base64-encoded audio
+            "description": "Audio file (e.g. mp3, wav) encoded as base64 string"
+        }
     }
 
     output_type = "string"
 
     def forward(self, audio: str) -> str:
         try:
-            decoded_audio = base64.b64decode(audio)
-            model = whisper.load_model("base")
-
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp:
-                tmp.write(decoded_audio)
-                tmp_path = tmp.name
+            audio_bytes = base64.b64decode(audio)
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
+                tmp_file.write(audio_bytes)
+                tmp_path = tmp_file.name
 
+            model = whisper.load_model("base")
             result = model.transcribe(tmp_path)
+
             os.remove(tmp_path)
             return result["text"]
+
         except Exception as e:
             return f"Transcription failed: {str(e)}"
 
 
 
+
 # --- Tool 3: Excel Reader Tool ---
-import base64
 import tempfile
-import pandas as pd
-import os
+import base64
 
 class ExcelReaderTool(Tool):
     name = "excel_reader"
-    description = "Reads uploaded Excel file (.xlsx) and summarizes the sheet data."
-    
-    parameters = {
-        "type": "object",
-        "properties": {
-            "excel": {
-                "type": "string",
-                "description": "The Excel file in base64-encoded format"
-            }
-        },
-        "required": ["excel"]
+    description = "Reads uploaded Excel file and summarizes sheet data."
+
+    inputs = {
+        "excel": {
+            "type": "string",  # base64-encoded .xlsx file
+            "description": "Excel file (.xlsx) encoded as base64 string"
+        }
     }
 
     output_type = "string"
 
     def forward(self, excel: str) -> str:
         try:
-            decoded_excel = base64.b64decode(excel)
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx") as tmp:
-                tmp.write(decoded_excel)
-                tmp_path = tmp.name
+            excel_bytes = base64.b64decode(excel)
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx") as tmp_file:
+                tmp_file.write(excel_bytes)
+                tmp_path = tmp_file.name
 
             df = pd.read_excel(tmp_path)
             os.remove(tmp_path)
-            return f"Summary:\n{df.describe(include='all').to_string()}"
+
+            return "Summary statistics:\n" + df.describe(include="all").to_string()
         except Exception as e:
             return f"Failed to read Excel file: {str(e)}"