agent with meme roaster tool

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, load_tool, tool
 import datetime
 import requests
 import pytz
@@ -7,16 +7,145 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
+# AI Framework Roasting Meme Generator Tool
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+def roast_ai_frameworks_meme(target_framework: str = "random", roast_style: str = "playful") -> str:
+    """Generates memes that playfully mock and roast AI agent frameworks and their quirks.
+    
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        target_framework: Which framework to roast (options: "langchain", "autogen", "crewai", "smolagents", "haystack", "random")
+        roast_style: How savage the roast should be (options: "playful", "savage", "brutal")
     """
-    return "What magic will you build ?"
+    
+    # Framework-specific roasts and stereotypes
+    framework_roasts = {
+        "langchain": {
+            "stereotypes": ["overcomplicated everything", "documentation nightmare", "callback hell", "abstraction overload"],
+            "common_issues": ["version breaking changes", "too many ways to do one thing", "memory management chaos", "dependency hell"],
+            "meme_scenarios": [
+                "trying to understand LangChain's architecture",
+                "reading LangChain docs for the 100th time",
+                "LangChain update breaking your entire project",
+                "explaining LangChain to a junior developer"
+            ]
+        },
+        "autogen": {
+            "stereotypes": ["agents talking in circles", "conversation loops", "Microsoft complexity", "over-engineered"],
+            "common_issues": ["agents arguing forever", "multi-agent chaos", "conversation management", "complex setup"],
+            "meme_scenarios": [
+                "AutoGen agents having a 3-hour argument about nothing",
+                "trying to stop AutoGen agents from talking",
+                "AutoGen agent conversation going completely off-topic",
+                "setting up AutoGen for a simple task"
+            ]
+        },
+        "crewai": {
+            "stereotypes": ["trying too hard to be cool", "crew member management", "workflow complexity", "startup vibes"],
+            "common_issues": ["crew coordination problems", "task delegation chaos", "role confusion", "process overhead"],
+            "meme_scenarios": [
+                "CrewAI agents not knowing their roles",
+                "crew members doing each other's jobs",
+                "CrewAI workflow taking longer than doing it manually",
+                "explaining why you need a 'crew' for a simple task"
+            ]
+        },
+        "smolagents": {
+            "stereotypes": ["'smol' but complicated", "trying to be minimalist", "tool management issues", "documentation gaps"],
+            "common_issues": ["tool integration problems", "limited ecosystem", "debugging nightmares", "unclear error messages"],
+            "meme_scenarios": [
+                "SmolagAgents being anything but 'smol'",
+                "trying to debug a SmolagAgents tool",
+                "SmolagAgents promising simplicity but delivering complexity",
+                "finding SmolagAgents documentation"
+            ]
+        },
+        "haystack": {
+            "stereotypes": ["search obsessed", "RAG everything", "pipeline confusion", "enterprise bloat"],
+            "common_issues": ["pipeline configuration hell", "version compatibility", "component confusion", "setup complexity"],
+            "meme_scenarios": [
+                "Haystack turning every problem into a search problem",
+                "configuring Haystack pipelines",
+                "Haystack's 47 different ways to do RAG",
+                "Haystack documentation sending you in circles"
+            ]
+        }
+    }
+    
+    # Select target framework
+    if target_framework == "random":
+        import random
+        target_framework = random.choice(list(framework_roasts.keys()))
+    
+    framework_data = framework_roasts.get(target_framework, framework_roasts["smolagents"])
+    
+    # Meme formats perfect for roasting
+    roast_formats = [
+        "Drake rejecting vs approving meme format",
+        "Expanding brain meme with increasingly absurd solutions",
+        "Distracted boyfriend looking at other frameworks",
+        "This is fine dog in burning room (framework on fire)",
+        "Galaxy brain ascending to framework enlightenment",
+        "Two buttons difficult choice between frameworks",
+        "Surprised Pikachu face when framework breaks",
+        "Woman yelling at cat (developer vs framework)",
+        "Stonks going down with framework complexity",
+        "Change my mind: framework is overcomplicated"
+    ]
+    
+    # Roast intensity levels
+    roast_levels = {
+        "playful": "gentle teasing, developer humor, lighthearted jokes about common issues",
+        "savage": "pointed criticism, highlighting real problems, witty burns about framework flaws", 
+        "brutal": "ruthless roasting, exposing all the pain points, maximum developer frustration humor"
+    }
+    
+    # Pick random elements
+    import random
+    selected_format = random.choice(roast_formats)
+    selected_scenario = random.choice(framework_data["meme_scenarios"])
+    selected_issue = random.choice(framework_data["common_issues"])
+    selected_stereotype = random.choice(framework_data["stereotypes"])
+    
+    roast_intensity = roast_levels[roast_style]
+    
+    # Create the roast meme prompt
+    meme_prompt = f"""Create a programming meme roasting {target_framework.title()} using {selected_format}.
+    
+    Scenario: {selected_scenario}
+    Focus on: {selected_issue} and {selected_stereotype}
+    
+    Roast style: {roast_intensity}
+    
+    The meme should be relatable to developers who have struggled with AI frameworks.
+    Include bold, readable meme text with typical internet meme formatting.
+    Make it funny but accurate to real developer experiences.
+    Use programming/developer culture references and inside jokes.
+    The visual should clearly convey developer frustration or irony.
+    Perfect for sharing on developer Twitter, Reddit, or Discord.
+    High contrast text (white with black outline) for maximum readability."""
+    
+    instructions = f"""
+🔥 AI FRAMEWORK ROAST MEME GENERATOR 🔥
+
+Target: {target_framework.title()}
+Roast Level: {roast_style.title()}
+Scenario: {selected_scenario}
+Format: {selected_format}
+
+ROAST PROMPT:
+{meme_prompt}
+
+BURN TOPICS:
+• Main Issue: {selected_issue}
+• Stereotype: {selected_stereotype}
+• Developer Pain Point: Framework complexity vs promised simplicity
+
+INSTRUCTIONS: Use the image generation tool with the prompt above to create this spicy developer meme!
+
+⚠️ WARNING: This meme may cause framework maintainers to cry and developers to nod in agreement! ⚠️
+"""
+    
+    return instructions
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +184,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, image_generation_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,