Update app.py

app.py

@@ -15,6 +15,7 @@ login(token=os.environ["HUGGINGFACEHUB_API_TOKEN"])
 ## TOOL 1
 search_tool = DuckDuckGoSearchTool()
 
+# --- Tool 1: Wikipedia Search Tool ---
 class WikipediaTool(Tool):
     name = "wikipedia_search"
     description = "Search Wikipedia for a given topic and return a concise summary."
@@ -36,9 +37,9 @@ class WikipediaTool(Tool):
             return "No Wikipedia page found."
         except Exception as e:
             return f"Error: {str(e)}"
-wiki_tool = WikipediaTool()
 
-# --- Tool 3: Audio Transcription Tool (uses whisper base) ---
+
+# --- Tool 2: Audio Transcription Tool (uses whisper base) ---
 class AudioTranscriptionTool(Tool):
     name = "audio_transcriber"
     description = "Transcribes an uploaded audio file to text using Whisper."
@@ -50,11 +51,11 @@ class AudioTranscriptionTool(Tool):
     }
     output_type = "string"
 
-    def forward(self, audio: str) -> str:
+    def forward(self, audio: bytes) -> str:
         try:
             model = whisper.load_model("base")
             with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
-                tmp_file.write(audio.read())
+                tmp_file.write(audio)
                 tmp_file_path = tmp_file.name
 
             result = model.transcribe(tmp_file_path)
@@ -62,25 +63,24 @@ class AudioTranscriptionTool(Tool):
             return result["text"]
         except Exception as e:
             return f"Transcription failed: {str(e)}"
-audio_tool = AudioTranscriptionTool()
+
 
 # --- Tool 3: Excel Reader Tool ---
 class ExcelReaderTool(Tool):
     name = "excel_reader"
     description = "Reads uploaded Excel file and summarizes sheet data."
     inputs = {
-    "excel": {
-        "type": "any",  # was "file"
-        "description": "The Excel (.xlsx) file to summarize"
+        "excel": {
+            "type": "any",
+            "description": "The Excel (.xlsx) file to summarize"
         }
     }
-
     output_type = "string"
 
-    def forward(self, excel) -> str:
+    def forward(self, excel: bytes) -> str:
         try:
             with tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx") as tmp_file:
-                tmp_file.write(excel.read())
+                tmp_file.write(excel)
                 tmp_file_path = tmp_file.name
 
             df = pd.read_excel(tmp_file_path)
@@ -89,7 +89,10 @@ class ExcelReaderTool(Tool):
             return f"Summary statistics:\n{summary}"
         except Exception as e:
             return f"Failed to read Excel file: {str(e)}"
+            
 excel_tool = ExcelReaderTool()
+wiki_tool = WikipediaTool()
+audio_tool = AudioTranscriptionTool()
 
 async def run_and_submit_all(profile: gr.OAuthProfile | None):
     log_output = ""