Upload 3 files

agent.py

@@ -0,0 +1,32 @@
+from tools import visit_webpage, image_generation_tool, image_diplay_tool
+from smolagents import (
+    CodeAgent,
+    ToolCallingAgent,
+    HfApiModel,
+    ManagedAgent,
+    DuckDuckGoSearchTool
+)
+
+def multi_agent_framework(model_id):
+    model = HfApiModel(model_id)
+
+    web_agent = ToolCallingAgent(
+        tools=[DuckDuckGoSearchTool(), visit_webpage],
+        model=model,
+        max_steps=5,
+    )
+
+    managed_web_agent = ManagedAgent(
+        agent=web_agent,
+        name="search",
+        description="Runs web searches for you. Give it your query as an argument.",
+    )
+
+    manager_agent = CodeAgent(
+        tools=[image_generation_tool, image_diplay_tool],
+        model=model,
+        managed_agents=[managed_web_agent],
+        additional_authorized_imports=["time", "numpy", "pandas", "requests"],
+    )
+
+    return manager_agent

app.py

@@ -0,0 +1,46 @@
+import streamlit as st
+from agent import multi_agent_framework
+
+
+# Define the multi agent framework
+model_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
+agent = multi_agent_framework(model_id)
+
+# Function to log agent actions
+def log_agent_action(prompt, result):
+    st.write(f"### Agent Activity")
+    st.write("**Prompt Sent to Agent:**")
+    st.code(prompt, language="text")
+    st.write("**Agent Output:**")
+    st.code(result, language="text")
+
+# Streamlit app title
+st.title("Multi Agent GPT")
+
+# App description
+st.write("Generate creative content, search the web and generate images enriched with the power of MultiAgent framework")
+
+# Input blog topic or prompt
+user_prompt = st.text_area("How may I help you?:", placeholder="E.g., Generate me a picture of cute puppy")
+
+# Button to generate content
+if st.button("Generate"):
+    if user_prompt:
+        with st.spinner("Generating content with our Multi agents"):
+            try:
+                # Run the agent with the given prompt
+                result = agent.run(user_prompt)
+                # Display the generated content
+                st.subheader("Generated Content:")
+                st.write(result)
+
+                # Log backend activity
+                log_agent_action(user_prompt, result)
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
+    else:
+        st.warning("Please enter a prompt to proceed.")
+
+# Footer
+st.markdown("---")
+st.caption("Powered by SmolAgents, DuckDuckGo, black-forest-labs and Streamlit")

tools.py

@@ -0,0 +1,61 @@
+import re
+import requests
+from PIL import Image
+from IPython.display import display
+from markdownify import markdownify
+from requests.exceptions import RequestException
+from smolagents import (
+    Tool,
+    tool
+)
+
+@tool
+def visit_webpage(url: str) -> str:
+    """Visits a webpage at the given URL and returns its content as a markdown string.
+
+    Args:
+        url: The URL of the webpage to visit.
+
+    Returns:
+        The content of the webpage converted to Markdown, or an error message if the request fails.
+    """
+    try:
+        # Send a GET request to the URL
+        response = requests.get(url)
+        response.raise_for_status()  # Raise an exception for bad status codes
+
+        # Convert the HTML content to Markdown
+        markdown_content = markdownify(response.text).strip()
+
+        # Remove multiple line breaks
+        markdown_content = re.sub(r"\n{3,}", "\n\n", markdown_content)
+
+        return markdown_content
+
+    except RequestException as e:
+        return f"Error fetching the webpage: {str(e)}"
+    except Exception as e:
+        return f"An unexpected error occurred: {str(e)}"
+
+@tool
+def image_diplay_tool(image_path : str) -> object:
+    """
+    This is a tool that returns the image object and displays it
+
+    Args:
+        image_path: The images's path for displaying.
+    """
+    try:
+        # Open the .webp image using Pillow
+        img = Image.open(image_path)
+        # Display the image
+        display(img)
+    except Exception as e:
+        print(f"Error displaying image: {e}")
+    return img
+
+image_generation_tool = Tool.from_space(
+    "black-forest-labs/FLUX.1-schnell",
+    name="image_generator",
+    description="Generate an image from a prompt and return its path. Make sure to improve the prompt. Another tool MUST be called for displaying image"
+)