Update app.py

app.py

@@ -6,40 +6,22 @@ from smolagents import CodeAgent, Tool
 from smolagents.models import HfApiModel
 from smolagents.monitoring import LogLevel
 from gradio import ChatMessage
-import logging
 from functools import lru_cache
 
-# Configure logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
 DEFAULT_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"
 HF_API_TOKEN = os.getenv("HF_TOKEN")
 
-# Tool descriptions for the UI
-TOOL_DESCRIPTIONS = {
-    "Hub Collections": "Add tool collections from Hugging Face Hub.",
-    "Spaces": "Add tools from Hugging Face Spaces.",
-}
-
 @lru_cache(maxsize=128)
 def search_spaces(query, limit=1):
-    """
-    Search for Hugging Face Spaces using the API.
-    Returns the first result or None if no results.
-    """
     try:
         url = f"https://huggingface.co/api/spaces?search={query}&limit={limit}"
-        response = requests.get(
-            url, headers={"Authorization": f"Bearer {HF_API_TOKEN}"}
-        )
+        response = requests.get(url, headers={"Authorization": f"Bearer {HF_API_TOKEN}"})
         response.raise_for_status()
         spaces = response.json()
         if not spaces:
             return None
         return extract_space_info(spaces[0])
-    except requests.RequestException as e:
-        logger.error(f"Error searching spaces: {e}")
+    except requests.RequestException:
         return None
 
 def extract_space_info(space):
@@ -57,33 +39,27 @@ def extract_space_info(space):
     return {"id": space_id, "title": title, "description": description}
 
 def get_space_metadata(space_id):
-    """
-    Get metadata for a specific Hugging Face Space.
-    """
     try:
         url = f"https://huggingface.co/api/spaces/{space_id}"
-        response = requests.get(
-            url, headers={"Authorization": f"Bearer {HF_API_TOKEN}"}
-        )
+        response = requests.get(url, headers={"Authorization": f"Bearer {HF_API_TOKEN}"})
         response.raise_for_status()
         space = response.json()
         return extract_space_info(space)
-    except requests.RequestException as e:
-        logger.error(f"Error getting space metadata: {e}")
+    except requests.RequestException:
         return None
 
 def create_agent(model_name, space_tools=None):
-    """
-    Create a CodeAgent with the specified model and tools.
-    """
     if not space_tools:
         space_tools = []
     try:
-        tools = [Tool.from_space(
-            tool_info["id"],
-            name=tool_info.get("name", tool_info["id"]),
-            description=tool_info.get("description", ""),
-        ) for tool_info in space_tools]
+        tools = [
+            Tool.from_space(
+                t["id"],
+                name=t.get("name", t["id"]),
+                description=t.get("description", ""),
+            )
+            for t in space_tools
+        ]
         model = HfApiModel(model_id=model_name, token=HF_API_TOKEN)
         agent = CodeAgent(
             tools=tools,
@@ -91,12 +67,9 @@ def create_agent(model_name, space_tools=None):
             additional_authorized_imports=["PIL", "requests"],
             verbosity_level=LogLevel.DEBUG,
         )
-        logger.info(f"Agent created successfully with {len(tools)} tools")
         return agent
-    except Exception as e:
-        logger.error(f"Error creating agent: {e}")
+    except:
         try:
-            logger.info("Trying fallback model...")
             fallback_model = HfApiModel(
                 model_id="Qwen/Qwen2.5-Coder-7B-Instruct", token=HF_API_TOKEN
             )
@@ -106,13 +79,10 @@ def create_agent(model_name, space_tools=None):
                 additional_authorized_imports=["PIL", "requests"],
                 verbosity_level=LogLevel.DEBUG,
             )
-            logger.info("Agent created successfully with fallback model")
             return agent
-        except Exception as e:
-            logger.error(f"Error creating agent with fallback model: {e}")
+        except:
             return None
 
-# Event handler functions
 def on_search_spaces(query):
     if not query:
         return "Please enter a search term.", "", "", ""
@@ -120,10 +90,13 @@ def on_search_spaces(query):
         space_info = search_spaces(query)
         if space_info is None:
             return "No spaces found.", "", "", ""
-        results_md = f"### Search Results:\n- ID: `{space_info['id']}`\n- Title: {space_info['title']}\n- Description: {space_info['description']}\n"
+        results_md = (
+            f"### Search Results:\n- ID: `{space_info['id']}`\n"
+            f"- Title: {space_info['title']}\n"
+            f"- Description: {space_info['description']}\n"
+        )
         return results_md, space_info["id"], space_info["title"], space_info["description"]
     except Exception as e:
-        logger.error(f"Error in search: {e}")
         return f"Error: {str(e)}", "", "", ""
 
 def on_validate_space(space_id):
@@ -135,12 +108,17 @@ def on_validate_space(space_id):
             space_info = search_spaces(space_id)
             if space_info is None:
                 return f"No spaces found for '{space_id}'.", "", ""
-            result_md = f"### Found Space via Search:\n- ID: `{space_info['id']}`\n- Title: {space_info['title']}\n- Description: {space_info['description']}\n"
+            result_md = (
+                f"### Found Space via Search:\n- ID: `{space_info['id']}`\n"
+                f"- Title: {space_info['title']}\n- Description: {space_info['description']}\n"
+            )
             return result_md, space_info["title"], space_info["description"]
-        result_md = f"### Space Validated Successfully:\n- ID: `{space_info['id']}`\n- Title: {space_info['title']}\n- Description: {space_info['description']}\n"
+        result_md = (
+            f"### Space Validated Successfully:\n- ID: `{space_info['id']}`\n"
+            f"- Title: {space_info['title']}\n- Description: {space_info['description']}\n"
+        )
         return result_md, space_info["title"], space_info["description"]
     except Exception as e:
-        logger.error(f"Error validating space: {e}")
         return f"Error: {str(e)}", "", ""
 
 def on_add_tool(space_id, space_name, space_description, current_tools):
@@ -157,7 +135,9 @@ def on_add_tool(space_id, space_name, space_description, current_tools):
     updated_tools = current_tools + [new_tool]
     tools_md = "### Added Tools:\n"
     for i, tool in enumerate(updated_tools, 1):
-        tools_md += f"{i}. **{tool['name']}** (`{tool['id']}`)\n   {tool['description']}\n\n"
+        tools_md += (
+            f"{i}. **{tool['name']}** (`{tool['id']}`)\n   {tool['description']}\n\n"
+        )
     return updated_tools, tools_md
 
 def on_create_agent(model, space_tools):
@@ -167,11 +147,10 @@ def on_create_agent(model, space_tools):
         agent = create_agent(model, space_tools)
         if agent is None:
             return None, [], "", "Failed to create agent. Please try again with different tools or model.", "No agent created yet."
-        tools_str = ", ".join([f"{tool['name']} ({tool['id']})" for tool in space_tools])
+        tools_str = ", ".join([f"{t['name']} ({t['id']})" for t in space_tools])
         agent_status = update_agent_status(agent)
         return agent, [], "", f"✅ Agent created successfully with {model}!\nTools: {tools_str}", agent_status
     except Exception as e:
-        logger.error(f"Error creating agent: {e}")
         return None, [], "", f"Error creating agent: {str(e)}", "No agent created yet."
 
 def add_user_message(message, chat_history):
@@ -217,11 +196,8 @@ def update_agent_status(agent):
     if agent is None:
         return "No agent created yet. Add a Space tool to get started."
     tools = agent.tools if hasattr(agent, "tools") else []
-    tool_count = len(tools)
-    status = f"Agent ready with {tool_count} tools"
-    return status
+    return f"Agent ready with {len(tools)} tools"
 
-# Create the Gradio app
 with gr.Blocks(title="AI Agent Builder") as app:
     gr.Markdown("# AI Agent Builder with smolagents")
     gr.Markdown("Build your own AI agent by selecting tools from Hugging Face Spaces.")
@@ -236,9 +212,20 @@ with gr.Blocks(title="AI Agent Builder") as app:
             model_input = gr.Textbox(value=DEFAULT_MODEL, label="Model", visible=False)
             with gr.Group():
                 gr.Markdown("### Add Space as Tool")
-                space_tool_input = gr.Textbox(label="Space ID or Search Term", placeholder=("Enter a Space ID or search term"), info="Enter a Space ID (username/space-name) or search term")
-                space_name_input = gr.Textbox(label="Tool Name (optional)", placeholder="Enter a name for this tool")
-                space_description_input = gr.Textbox(label="Tool Description (optional)", placeholder="Enter a description for this tool", lines=2)
+                space_tool_input = gr.Textbox(
+                    label="Space ID or Search Term",
+                    placeholder="Enter a Space ID (username/space-name) or search term",
+                    info="Enter a Space ID (username/space-name) or search term"
+                )
+                space_name_input = gr.Textbox(
+                    label="Tool Name (optional)",
+                    placeholder="Enter a name for this tool"
+                )
+                space_description_input = gr.Textbox(
+                    label="Tool Description (optional)",
+                    placeholder="Enter a description for this tool",
+                    lines=2
+                )
                 add_tool_button = gr.Button("Add Tool", variant="primary")
             gr.Markdown("### Added Tools")
             tools_display = gr.Markdown("No tools added yet. Add at least one tool before creating an agent.")
@@ -256,7 +243,9 @@ with gr.Blocks(title="AI Agent Builder") as app:
     space_tool_input.submit(on_validate_space, inputs=[space_tool_input], outputs=[status_msg, space_name_input, space_description_input])
     add_tool_button.click(on_add_tool, inputs=[space_tool_input, space_name_input, space_description_input, space_tools_state], outputs=[space_tools_state, tools_display])
     create_button.click(on_create_agent, inputs=[model_input, space_tools_state], outputs=[agent_state, chatbot, msg, status_msg, agent_status])
-    msg.submit(lambda message: (message, message, ""), inputs=[msg], outputs=[msg_store, msg, msg], queue=False).then(add_user_message, inputs=[msg_store, chatbot], outputs=[msg_store, chatbot], queue=False).then(stream_agent_response, inputs=[agent_state, msg_store, chatbot], outputs=chatbot, queue=True)
+    msg.submit(lambda message: (message, message, ""), inputs=[msg], outputs=[msg_store, msg, msg], queue=False)\
+       .then(add_user_message, inputs=[msg_store, chatbot], outputs=[msg_store, chatbot], queue=False)\
+       .then(stream_agent_response, inputs=[agent_state, msg_store, chatbot], outputs=chatbot, queue=True)
 
 if __name__ == "__main__":
     app.queue().launch()