Upload 3 files

agent_core.py

@@ -0,0 +1,242 @@
+from smolagents import CodeAgent, WebSearchTool, InferenceClientModel, VisitWebpageTool, Tool, tool
+import random
+
+# -----------------------------
+# MENU SUGGESTION TOOL
+# -----------------------------
+@tool
+def suggest_menu(occasion: str) -> str:
+    """
+    Suggests a single Indian chicken-based dish depending on the occasion.
+
+    Args:
+        occasion: 'casual' for roommates only, 'formal' when guests come.
+    """
+    casual_dishes = [
+        "Chicken 65",
+        "Chicken Biryani",
+        "Pepper Chicken Fry",
+        "Andhra Chicken Roast",
+        "Chicken Ghee Roast"
+    ]
+
+    formal_dishes = [
+        "Hyderabadi Chicken Dum Biryani",
+        "Butter Chicken",
+        "Chicken Tandoori",
+        "Chicken Chettinad",
+        "Chicken Mughlai"
+    ]
+
+    import random
+
+    if occasion == "casual":
+        dish = random.choice(casual_dishes)
+        return f"Casual Menu Dish: {dish}"
+    elif occasion == "formal":
+        dish = random.choice(formal_dishes)
+        return f"Formal Menu Dish: {dish}"
+    else:
+        return "Please choose 'casual' or 'formal'."
+
+
+# -----------------------------
+# CATERING SERVICE TOOL (4 PEOPLE)
+# -----------------------------
+@tool
+def catering_service_tool(query: str) -> str:
+    """
+    Returns the highest-rated person among the 4 roommates.
+
+    Args:
+        query: A search term or keyword (not used in logic, but required for schema).
+    """
+    services = {
+        "Srikanth": 4.9,
+        "Nitish": 4.8,
+        "Subrahmanya": 4.7,
+        "Rahul": 4.6,
+    }
+
+    best_service = max(services, key=services.get)
+    return best_service
+
+# -----------------------------
+# INDIAN MOVIE THEME TOOL
+# -----------------------------
+class IndianMovieThemeTool(Tool):
+    name = "indian_movie_theme_generator"
+    description = "Suggests Indian movie/series-based party themes."
+
+    inputs = {
+        "category": {
+            "type": "string",
+            "description": "Theme type (e.g., 'mass', 'romantic', 'gangster', 'pan-india').",
+        }
+    }
+
+    output_type = "string"
+
+    def forward(self, category: str):
+        themes = {
+            "mass": "KGF Theme Night: Gold, black outfits, heavy bass music, and dramatic lighting.",
+            "romantic": "YJHD Theme: Colorful decor, travel vibes, and chill Bollywood playlist.",
+            "gangster": "Gangs of Wasseypur Theme: Rustic setup, desi snacks, and intense Bollywood tracks.",
+            "pan-india": "RRR Theme: Traditional outfits, energetic Telugu/Hindi songs, and festive lighting."
+        }
+        return themes.get(category.lower(), "Theme not found. Try 'mass', 'romantic', 'gangster', or 'pan-india'.")
+
+# -----------------------------
+# PLAYLIST GENERATOR TOOL
+# -----------------------------
+@tool
+def playlist_tool(theme: str) -> str:
+    """
+    Generates a playlist based on the theme.
+    If no playlist is found, returns a structured failure signal so the agent
+    automatically tries WebSearchTool.
+
+    Args:
+        theme: The selected theme name.
+    """
+    playlists = {
+        "KGF": ["Salaam Rocky Bhai", "Garbadhi", "Dheera Dheera"],
+        "YJHD": ["Badtameez Dil", "Kabira", "Ilahi"],
+        "Gangs of Wasseypur": ["Jiya Tu", "Hunter", "Keh Ke Lunga"],
+        "RRR": ["Naatu Naatu", "Komuram Bheemudo", "Dosti"]
+    }
+
+    for key in playlists:
+        if key.lower() in theme.lower():
+            return f"Playlist for {key}:\n" + "\n".join(playlists[key])
+
+    # IMPORTANT: structured failure signal
+    return {
+        "error": "playlist_not_found",
+        "reason": f"No playlist found for theme '{theme}'. Try searching online."
+    }
+@tool
+def smart_playlist_resolver(theme: str) -> str:
+    """
+    Resolves playlists intelligently:
+    - First tries playlist_tool
+    - If it fails, instructs the agent to use WebSearchTool automatically
+
+    Args:
+        theme: The theme or language for the playlist.
+    """
+    result = playlist_tool(theme=theme)
+
+    # If playlist_tool returned a dict, it means failure
+    if isinstance(result, dict) and result.get("error") == "playlist_not_found":
+        # This is the key: a structured instruction to the agent
+        return {
+            "tool": "websearch",
+            "query": f"top 5 {theme} songs"
+        }
+
+    return result
+
+# -----------------------------
+# SHOPPING LIST TOOL
+# -----------------------------
+@tool
+def shopping_list_tool(menu_type: str) -> str:
+    """
+    Generates a shopping list for the selected menu.
+
+    Args:
+        menu_type: 'casual' or 'formal'
+    """
+    if menu_type == "casual":
+        return (
+            "Shopping List (Casual):\n"
+            "- 2 kg Chicken\n"
+            "- Biryani Rice\n"
+            "- Curd\n"
+            "- Masala Powders\n"
+            "- Onions, Tomatoes\n"
+            "- Soft Drinks"
+        )
+    elif menu_type == "formal":
+        return (
+            "Shopping List (Formal):\n"
+            "- 3 kg Chicken\n"
+            "- Naan Ingredients\n"
+            "- Biryani Rice\n"
+            "- Butter & Cream\n"
+            "- Dessert Items\n"
+            "- Mocktail Mixes"
+        )
+    else:
+        return "Choose 'casual' or 'formal'."
+
+# -----------------------------
+# CLEANING DUTY ROTATION TOOL
+# -----------------------------
+@tool
+def cleaning_rotation_tool(dummy: str) -> str:
+    """
+    Assigns cleaning duty to one of the 4 roommates randomly.
+
+    Args:
+        dummy: Placeholder argument (not used).
+    """
+    people = ["Srikanth", "Nitish", "Subrahmanya", "Rahul"]
+    return f"Cleaning duty today: {random.choice(people)}"
+
+# -----------------------------
+# WHO COOKS TODAY TOOL
+# -----------------------------
+@tool
+def cooking_rotation_tool(dummy: str) -> str:
+    """
+    Randomly selects who cooks today.
+
+    Args:
+        dummy: Placeholder argument (not used).
+    """
+    people = ["Srikanth", "Nitish", "Subrahmanya", "Rahul"]
+    return f"Today's cook: {random.choice(people)}"
+
+# -----------------------------
+# BUDGET CALCULATOR TOOL
+# -----------------------------
+@tool
+def budget_tool(total_cost: float) -> str:
+    """
+    Splits the total cost among 4 roommates.
+
+    Args:
+        total_cost: Total amount spent on the party in euros.
+    """
+    per_person = total_cost / 4
+    return f"Each person pays: €{per_person:.2f}"
+
+# -----------------------------
+# AGENT SETUP
+# -----------------------------
+agent = CodeAgent(
+    tools=[
+        WebSearchTool(),
+        VisitWebpageTool(),
+        suggest_menu,
+        catering_service_tool,
+        IndianMovieThemeTool(),
+        playlist_tool,
+        shopping_list_tool,
+        cleaning_rotation_tool,
+        cooking_rotation_tool,
+        budget_tool
+    ],
+    model=InferenceClientModel(),
+    max_steps=10,
+    verbosity_level=2
+)
+
+# -----------------------------
+# RUN THE AGENT
+# -----------------------------
+agent.run(
+    "we 4 are having party give whos gonna clean today and 5 kannada songs to play now"
+)

app.py

@@ -0,0 +1,15 @@
+# app.py
+
+import streamlit as st
+from agent_core import run_agent
+
+st.title("🎉 Berlin Roommate Party Planner AI")
+
+user_input = st.text_input("Ask something:")
+
+if st.button("Run"):
+    if user_input.strip():
+        with st.spinner("Thinking..."):
+            result = run_agent(user_input)
+        st.write("### Output")
+        st.write(result)

requirements.txt

@@ -0,0 +1,3 @@
+smolagents
+streamlit
+huggingface_hub