Update app.py

app.py

@@ -1,10 +1,8 @@
-import json
-import contextlib
 import mimetypes
 import os
 import re
 import shutil
-from typing import Dict, List, Optional, Any
+from typing import Optional
 
 from dotenv import load_dotenv
 from huggingface_hub import login
@@ -80,13 +78,14 @@ ALLOWED_FILE_TYPES = [
     "text/plain",
     "image/png",
     "image/jpeg",  # Added JPEG support
-    "image/gif",   # Added GIF support
+    "image/gif",  # Added GIF support
     "video/mp4",
     "audio/mpeg",  # Added MP3 support
-    "audio/wav",   # Added WAV support
-    "audio/ogg",   # Added OGG support
+    "audio/wav",  # Added WAV support
+    "audio/ogg",  # Added OGG support
 ]
 
+
 def setup_environment():
     """Initialize environment variables and authentication."""
     load_dotenv(override=True)
@@ -96,6 +95,7 @@ def setup_environment():
     else:
         print("HF_TOKEN not found in environment variables.")
 
+
 # ------------------------ Model and Tool Management ------------------------
 class ModelManager:
     """Manages model loading and initialization."""
@@ -135,6 +135,7 @@ class ModelManager:
             print(f"✗ Couldn't load model: {e}")
             raise
 
+
 class ToolRegistry:
     """Manages tool initialization and organization."""
 
@@ -178,6 +179,7 @@ class ToolRegistry:
             print(f"✗ Couldn't initialize image generation tool: {e}")
             raise
 
+
 # ------------------------ Agent Creation and Execution ------------------------
 def create_agent():
     """Creates a fresh agent instance with properly configured tools."""
@@ -214,6 +216,7 @@ def create_agent():
         planning_interval=4,
     )
 
+
 def stream_to_gradio(
     agent,
     task: str,
@@ -239,18 +242,19 @@ def stream_to_gradio(
     elif isinstance(final_answer, AgentImage):
         yield gr.ChatMessage(
             role="assistant",
-            content= { "image": final_answer.to_string(), "type": "file" },
+            content={"image": final_answer.to_string(), "type": "file"},
         )  # Send as Gradio-compatible file object:
     elif isinstance(final_answer, AgentAudio):
         yield gr.ChatMessage(
             role="assistant",
-            content={ "audio": final_answer.to_string(), "type": "file" },
-        ) # Send as Gradio-compatible file object
+            content={"audio": final_answer.to_string(), "type": "file"},
+        )  # Send as Gradio-compatible file object
     else:
         yield gr.ChatMessage(
             role="assistant", content=f"**Final answer:** {str(final_answer)}"
         )
 
+
 # ------------------------ Gradio UI Components ------------------------
 class GradioUI:
     """A one-line interface to launch your agent in Gradio."""
@@ -327,11 +331,16 @@ class GradioUI:
         sanitized_name = "".join(name_parts) + extension
 
         # Limit File Size, and Throw Error
-        max_file_size_mb = 50 # Define the limit
+        max_file_size_mb = 50  # Define the limit
         file_size_mb = os.path.getsize(file.name) / (1024 * 1024)  # Size in MB
 
         if file_size_mb > max_file_size_mb:
-            return gr.Textbox(f"File size exceeds {max_file_size_mb} MB limit.", visible=True), file_uploads_log
+            return (
+                gr.Textbox(
+                    f"File size exceeds {max_file_size_mb} MB limit.", visible=True
+                ),
+                file_uploads_log,
+            )
 
         # Save the uploaded file to the specified folder
         file_path = os.path.join(self.file_upload_folder, sanitized_name)
@@ -353,7 +362,7 @@ class GradioUI:
             gr.Textbox(
                 value="",
                 interactive=False,
-                placeholder="Processing...", # Changed placeholder.
+                placeholder="Processing...",  # Changed placeholder.
             ),
             gr.Button(interactive=False),
         )
@@ -400,7 +409,9 @@ class GradioUI:
                 else:
                     return self._create_mobile_layout()
 
-        demo.queue(max_size=20).launch(debug=True, **kwargs)  # Add queue with reasonable size
+        demo.queue(max_size=20).launch(
+            debug=True, **kwargs
+        )  # Add queue with reasonable size
 
     def _create_desktop_layout(self):
         """Create the desktop layout with sidebar."""
@@ -575,6 +586,7 @@ class GradioUI:
             [text_input, launch_research_btn],
         )
 
+
 # ------------------------ Execution ------------------------
 def main():
     """Main entry point for the application."""
@@ -587,5 +599,6 @@ def main():
     # Launch UI
     GradioUI(file_upload_folder="uploaded_files").launch()
 
+
 if __name__ == "__main__":
     main()