Update app.py

app.py

@@ -7,10 +7,11 @@ import re
 import shutil
 from typing import Optional
 
+import gradio as gr
 from dotenv import load_dotenv
 from huggingface_hub import login
-import gradio as gr
-
+from scripts.flux_lora_tool import FluxLoRATool
+from scripts.text_cleaner_tool import TextCleanerTool
 from scripts.text_inspector_tool import TextInspectorTool
 from scripts.text_web_browser import (
     ArchiveSearchTool,
@@ -22,20 +23,17 @@ from scripts.text_web_browser import (
     VisitTool,
 )
 from scripts.visual_qa import visualizer
-from scripts.text_cleaner_tool import TextCleanerTool
-from scripts.flux_lora_tool import FluxLoRATool
-
 from smolagents import (
     CodeAgent,
+    GoogleSearchTool,
     HfApiModel,
     LiteLLMModel,
     OpenAIServerModel,
-    TransformersModel,
-    GoogleSearchTool,
     Tool,
+    TransformersModel,
 )
-from smolagents.agent_types import AgentText, AgentImage, AgentAudio
-from smolagents.gradio_ui import pull_messages_from_step, handle_agent_output_types
+from smolagents.agent_types import AgentAudio, AgentImage, AgentText
+from smolagents.gradio_ui import handle_agent_output_types, pull_messages_from_step
 
 # ------------------------ Configuration and Setup ------------------------
 # Constants and configurations
@@ -144,13 +142,13 @@ class ModelManager:
             if chosen_inference == "hf_api":
                 return HfApiModel(model_id=model_id)
 
-            elif chosen_inference == "hf_api_provider":
+            if chosen_inference == "hf_api_provider":
                 return HfApiModel(provider="together")
 
-            elif chosen_inference == "litellm":
+            if chosen_inference == "litellm":
                 return LiteLLMModel(model_id=model_id)
 
-            elif chosen_inference == "openai":
+            if chosen_inference == "openai":
                 if not key_manager:
                     raise ValueError("Key manager required for OpenAI model")