Update app.py

app.py

@@ -1,112 +1,69 @@
-import streamlit as st
-import os
-import glob
-import re
-import base64
-import pytz
-from datetime import datetime
+import gradio as gr
 from transformers.agents import CodeAgent, ReactCodeAgent, ReactJsonAgent, load_tool
+import random
 
-# Page config
-st.set_page_config(page_title="🌳✨ AI Knowledge Tree Builder 🛠️🤓", page_icon="🌳", layout="wide")
-st.sidebar.markdown("""
-🌳🤖 **AI Knowledge Tree Builder**
-1. 📱 Universal access
-2. ⚡ Rapid builds (<2min)
-3. 🔗 Linked AI sources
-4. 🎯 Lean core
-5. 🧠 Shareable memory
-6. 👤 Personalized
-7. 🐦 Cloneable brevity
-""")
+# Custom tools since some built-ins are flaky
+class WittyQuoteTool:
+    def __init__(self):
+        self.name = "witty_quote"
+        self.description = "Spouts a random witty ML quote! 😂"
+        self.inputs = {}
+        self.output_type = str
 
-# Initialize Agents
-tools = [load_tool("text-to-speech"), load_tool("image_question_answering")]
-agents = {
-    "CodeCrafter": CodeAgent(tools=tools, system_prompt="Craft code like a pro! 🖥️"),
-    "StepSage": ReactCodeAgent(tools=tools, system_prompt="Step-by-step wisdom! 🧠"),
-    "JsonJugger": ReactJsonAgent(tools=tools, system_prompt="JSON-powered antics! 🤡"),
-    "OutlineOracle": ReactCodeAgent(tools=tools, system_prompt="Outline everything! 📋"),
-    "ToolTitan": CodeAgent(tools=tools, system_prompt="List tools with swagger! 🔧"),
-    "SpecSpinner": ReactJsonAgent(tools=tools, system_prompt="Spin specs with style! 📜"),
-    "ImageImp": CodeAgent(tools=tools, system_prompt="Dream up image prompts! 🎨"),
-    "VisualVortex": ReactCodeAgent(tools=tools, system_prompt="Generate visuals! 🖼️"),
-    "GlossGuru": ReactJsonAgent(tools=tools, system_prompt="Define terms with pizzazz! 📖"),
-}
-
-# MoE Prompts mapped to agents
-moe_prompts = {
-    "Create a python streamlit app.py...": "CodeCrafter",
-    "Create a python gradio app.py...": "CodeCrafter",
-    "Create a mermaid model...": "OutlineOracle",
-    "Create a top three list of tools...": "ToolTitan",
-    "Create a specification in markdown...": "SpecSpinner",
-    "Create an image generation prompt...": "ImageImp",
-    "Generate an image which describes...": "VisualVortex",
-    "List top ten glossary terms...": "GlossGuru",
-    "": "StepSage"  # Default agent for blank selection
-}
+    def __call__(self):
+        quotes = ["'AI is just spicy statistics' - Unknown 🤓", "'The only limit is your GPU budget' - Me 💸"]
+        return random.choice(quotes)
 
-# Session state
-if 'selected_file' not in st.session_state:
-    st.session_state.selected_file = None
-if 'view_mode' not in st.session_state:
-    st.session_state.view_mode = 'view'
-if 'files' not in st.session_state:
-    st.session_state.files = []
+class MLDevTool:
+    def __init__(self):
+        self.name = "ml_dev_helper"
+        self.description = "Helps with ML dev tasks like coding or debugging! 🛠️"
+        self.inputs = {"task": {"type": "str", "description": "What ML task to assist with"}}
+        self.output_type = str
 
-# MoE Selection
-moe_options = list(moe_prompts.keys())
-selected_moe = st.selectbox("Pick an MoE Adventure! 🎲", moe_options, index=0, key="selected_moe")
+    def __call__(self, task):
+        return f"🧠 Beep boop! Helping with: {task}. Tip: Always check your data first! 📊"
 
-# Utility Functions (simplified for brevity, reusing yours where unchanged)
-def sanitize_filename(text):
-    return re.sub(r'[^\w\s-]', ' ', text.strip())[:50]
+# Initialize 9 agents with stable tools
+tools = [WittyQuoteTool(), MLDevTool()]  # Dropped problematic built-ins
+try:
+    tools.append(load_tool("speech_to_text"))  # Safer built-in option
+except Exception as e:
+    print(f"Skipping speech_to_text due to: {e}")
 
-def generate_timestamp_filename(query):
-    central = pytz.timezone('US/Central')
-    now = datetime.now(central)
-    return f"{now.strftime('%I%M%p %m%d%Y')} ({sanitize_filename(query)}).md"
-
-def save_ai_interaction(query, result, is_rerun=False):
-    filename = generate_timestamp_filename(query)
-    content = f"# {'Rerun' if is_rerun else 'Query'}: {query}\n\n## AI Response\n{result}"
-    with open(filename, 'w', encoding='utf-8') as f:
-        f.write(content)
-    return filename
-
-def run_agent(task, agent_name):
-    agent = agents[agent_name]
-    if isinstance(agent, CodeAgent):
-        return agent.run(task, return_generated_code=True)
-    return agent.run(task)
+agents = {
+    "CodeWizard": CodeAgent(tools=tools, system_prompt="You're a coding genius! 🧙‍♂️ Write concise code."),
+    "StepByStep": ReactCodeAgent(tools=tools, system_prompt="Think step-by-step, oh wise one! 🤖"),
+    "JsonJester": ReactJsonAgent(tools=tools, system_prompt="JSON all the things! 😂"),
+    "DataDiva": ReactCodeAgent(tools=tools, system_prompt="Clean that data, superstar! 🌟"),
+    "ModelMaestro": CodeAgent(tools=tools, system_prompt="Tune models like a pro! 🎻"),
+    "DebugDemon": ReactJsonAgent(tools=tools, system_prompt="Squash bugs with flair! 👾"),
+    "DocDynamo": ReactCodeAgent(tools=tools, system_prompt="Write docs that dazzle! 📜"),
+    "VizVirtuoso": CodeAgent(tools=tools, system_prompt="Visualize like a boss! 📈"),
+    "DeployDaredevil": ReactJsonAgent(tools=tools, system_prompt="Deploy with guts! 🚀"),
+}
 
-# Main App
-def main():
-    st.markdown("### 🌳 AI Knowledge Tree Builder 🧠🌱 Let’s Grow Some Smarts!")
-    query = st.text_input("Ask Away! 🤔", placeholder="E.g., 'Explain transformers!'")
-    
-    if query:
-        agent_name = moe_prompts[selected_moe]
+def run_ml_team(task):
+    outputs = []
+    for name, agent in agents.items():
         try:
-            result = run_agent(f"{selected_moe} {query}" if selected_moe else query, agent_name)
-            st.markdown(f"🎉 {agent_name} says: {result}")
-            saved_file = save_ai_interaction(query, result)
-            st.success(f"Saved to {saved_file}")
-            st.session_state.selected_file = saved_file
+            if isinstance(agent, CodeAgent):
+                result = agent.run(task, return_generated_code=True)
+                outputs.append(f"🎉 {name} says: Here's my code magic! ```python\n{result}\n```")
+            else:
+                result = agent.run(task)
+                outputs.append(f"🔥 {name} reacts: {result}")
         except Exception as e:
-            st.error(f"😱 Oops! {e}")
+            outputs.append(f"😱 {name} tripped over a wire: {str(e)}")
+    return "\n\n".join(outputs)
 
-    # File Management (simplified)
-    st.sidebar.title("📁 Files")
-    md_files = sorted([f for f in glob.glob("*.md") if f.lower() != 'readme.md'])
-    for file in md_files:
-        st.sidebar.write(file)
-        if st.sidebar.button(f"View {file}", key=f"v_{file}"):
-            st.session_state.selected_file = file
-    if st.session_state.selected_file:
-        with open(st.session_state.selected_file, 'r') as f:
-            st.markdown(f.read())
+# Gradio UI
+with gr.Blocks(title="ML Agent Party 🎉") as demo:
+    gr.Markdown("# ML Agent Party! 🤖🎈\n9 Agents to Boost Your ML Game with Sass & Smarts!")
+    task_input = gr.Textbox(label="What's Your ML Quest? 🗡️", placeholder="E.g., 'Clean my data!'")
+    output = gr.Markdown(label="Agent Squad Response 🎤")
+    submit_btn = gr.Button("Unleash the Crew! 🚀")
+    submit_btn.click(run_ml_team, inputs=task_input, outputs=output)
+    gr.Markdown("---\nBuilt with ❤️ & 😂 by xAI's Grok & Transformers Agents")
 
-if __name__ == "__main__":
-    main()
+demo.launch()