Update app.py

app.py

@@ -1,304 +1,346 @@
 #!/usr/bin/env python3
 """
-You've Got Creed - Gradio Edition
-Because Creed Bratton + AI = Pure Chaos
-
-Optimized for GPU Zero - Loads instantly!
+🎸 Creed Bratton AI with MCP Support - Now Claude can use Creed as a tool!
+Built for maximum chaos and enterprise AI integration
 """
 
+import os
 import gradio as gr
-import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer
-from peft import PeftModel
-import time
-import re
 import random
+import time
+from typing import List, Tuple, Iterator
 
-class CreedAI:
-    def __init__(self):
-        self.model = None
-        self.tokenizer = None
-        self.device = "cuda" if torch.cuda.is_available() else "cpu"
-        self.load_model()
-        
-    def load_model(self):
-        """Load Creed - should be instant with 0.5B"""
-        try:
-            print("🧠 Loading Creed's consciousness...")
-            
-            # Load base model
-            base_model = AutoModelForCausalLM.from_pretrained(
-                "Qwen/Qwen2.5-0.5B",
-                torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-                device_map="auto" if torch.cuda.is_available() else None,
-                trust_remote_code=True
-            )
-            
-            # Add LoRA adapter
-            self.model = PeftModel.from_pretrained(
-                base_model,
-                "phxdev/creed-qwen-0.5b-lora",
-                torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
-            )
-            
-            # Load tokenizer
-            self.tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-0.5B")
-            if self.tokenizer.pad_token is None:
-                self.tokenizer.pad_token = self.tokenizer.eos_token
-                
-            self.model.eval()
-            print("✅ Creed is ready!")
-            
-        except Exception as e:
-            print(f"❌ Error loading Creed: {str(e)}")
-            self.model = None
+# Disable SSR to prevent JavaScript scroll errors
+os.environ['GRADIO_SSR_MODE'] = 'false'
 
-    def chat(self, message, history, show_raw_stream=True, show_thinking=True, temperature=0.8, max_tokens=512):
-        """Chat with Creed - with raw consciousness stream"""
-        if not self.model:
-            return history + [["❌ Error", "Creed couldn't load. The computers are against me again."]]
-            
-        try:
-            # Build context
-            context = self._build_context(message, history)
-            
-            inputs = self.tokenizer(
-                context,
-                return_tensors="pt",
-                truncation=True,
-                max_length=1024
-            ).to(self.device)
-            
-            # Generate
-            with torch.no_grad():
-                outputs = self.model.generate(
-                    **inputs,
-                    max_new_tokens=max_tokens,
-                    temperature=temperature,
-                    do_sample=True,
-                    top_p=0.9,
-                    top_k=50,
-                    pad_token_id=self.tokenizer.pad_token_id,
-                    eos_token_id=self.tokenizer.eos_token_id,
-                    repetition_penalty=1.1
-                )
-            
-            # Decode raw response
-            full_response = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
-            new_content = full_response[len(context):].strip()
-            
-            # Show different views based on settings
-            if show_raw_stream:
-                # Raw consciousness stream - no parsing, no filtering
-                formatted_response = f"🧠 **RAW CREED CONSCIOUSNESS:**\n```\n{new_content}\n```"
-                
-                if show_thinking:
-                    # Also show parsed version below
-                    thinking, response = self._parse_response(new_content)
-                    if thinking:
-                        formatted_response += f"\n\n🎸 **Parsed thoughts:**\n_{thinking}_\n\n**Clean response:** {response}"
-                    else:
-                        formatted_response += f"\n\n🎸 **Clean response:** {response}"
-            else:
-                # Traditional parsed response
-                thinking, response = self._parse_response(new_content)
-                if show_thinking and thinking:
-                    formatted_response = f"🧠 **Creed's thoughts:**\n_{thinking}_\n\n🎸 **Creed:** {response}"
-                else:
-                    formatted_response = response
-            
-            # Return updated history
-            return history + [[message, formatted_response]]
-            
-        except Exception as e:
-            error_response = f"❌ Error: {str(e)}\n\n🌱 *The mung beans are interfering with the computers again...*"
-            return history + [[message, error_response]]
-
-    def _build_context(self, user_input, history):
-        """Build conversation context"""
-        context_lines = []
-        
-        # Add recent history (last 3 exchanges)
-        for human_msg, ai_msg in history[-3:]:
-            context_lines.append(f"Human: {human_msg}")
-            # Clean AI response for context
-            clean_ai = re.sub(r'🧠.*?\n\n🎸.*?\*\*|🎸.*?\*\*', '', ai_msg)
-            clean_ai = re.sub(r'_.*?_', '', clean_ai).strip()
-            context_lines.append(f"Assistant: {clean_ai}")
+def creed_wisdom_tool(topic: str = "life") -> str:
+    """
+    Get profound wisdom from Creed Bratton about any topic.
+    
+    Args:
+        topic (str): The topic you want Creed's wisdom about
         
-        # Add current input
-        context_lines.append(f"Human: {user_input}")
-        context_lines.append("Assistant:")
+    Returns:
+        str: Creed's unique perspective on the topic
+    """
+    
+    wisdom_responses = {
+        "life": [
+            "Sometimes a guy's gotta ride the bull, am I right? Later, skater.",
+            "The only difference between me and a homeless man is this job. I will do whatever it takes to survive.",
+            "I've been involved in a number of cults, both as a leader and a follower. You have more fun as a follower, but you make more money as a leader."
+        ],
+        "business": [
+            "I run a small fake ID company from my car with a laminating machine that I swiped from the Sheriff's station.",
+            "Nobody steals from Creed Bratton and gets away with it. The last person to do this disappeared. His name? Creed Bratton.",
+            "Oh, you're paying way too much for worms, man. Who's your worm guy?"
+        ],
+        "relationships": [
+            "In the '60s, I made love to many, many women, often outdoors, in the mud and the rain... and it's possible a man slipped in.",
+            "I'm not offended by homosexuality. In the '60s, I made love to many, many women, often outdoors, in the mud and the rain.",
+            "Cool beans, man. I live by the quarry. We should hang out by the quarry and throw things down there!"
+        ],
+        "career": [
+            "I sprouted mung beans on a damp paper towel in my desk drawer. Very nutritious, but they smell like death.",
+            "Every week, I'm supposed to take four hours and do a quality spot-check at the paper mill. And of course the one year I blow it off, this happens.",
+            "When Pam gets Michael's old chair, I get Pam's old chair. Then I'll have two chairs. Only one to go."
+        ]
+    }
+    
+    # Get wisdom for the topic, or default to life advice
+    topic_wisdom = wisdom_responses.get(topic.lower(), wisdom_responses["life"])
+    response = random.choice(topic_wisdom)
+    
+    # Add some random Creed flair
+    if random.random() < 0.4:
+        extra_creed = [
+            "Also, I once arm-wrestled a bear.",
+            "The moon is made of cheese, obviously.",
+            "I can speak fluent dolphin.",
+            "My middle name is Danger.",
+            "I invented the question mark."
+        ]
+        response += f" {random.choice(extra_creed)}"
+    
+    return response
+
+def creed_story_generator(mood: str = "mysterious") -> str:
+    """
+    Generate a classic Creed Bratton story based on mood.
+    
+    Args:
+        mood (str): The mood for the story (mysterious, crazy, profound, random)
         
-        return "\n".join(context_lines)
+    Returns:
+        str: A vintage Creed story
+    """
+    
+    stories = {
+        "mysterious": [
+            "You were in the parking lot earlier! That's how I know you!",
+            "www.creedthoughts.gov.www\\creedthoughts",
+            "Strike, scream, run."
+        ],
+        "crazy": [
+            "Two eyes, two ears, a chin, a mouth, ten fingers, two nipples, a butt, two kneecaps, a penis. I have just described to you the Loch Ness Monster.",
+            "If I can't scuba, then what's this all been about? What have I been working toward?",
+            "The Taliban is the worst. Great heroin though."
+        ],
+        "profound": [
+            "I already won the lottery. I was born in the US of A, baby!",
+            "I don't technically have a hearing problem, but sometimes when there's a lot of noises occurring at the same time, I'll hear 'em as one big jumble.",
+            "I'm 30. Well, in November I'll be 30."
+        ]
+    }
+    
+    mood_stories = stories.get(mood.lower(), stories["mysterious"])
+    return random.choice(mood_stories)
 
-    def _parse_response(self, content):
-        """Parse thinking process from response"""
-        thinking = ""
-        response = content
+def creed_business_advice(situation: str) -> str:
+    """
+    Get business advice from Creed Bratton's entrepreneurial experience.
+    
+    Args:
+        situation (str): Describe your business situation
         
-        # Look for thinking patterns
-        thinking_match = re.search(r'<thinking>(.*?)</thinking>', content, re.DOTALL)
-        if thinking_match:
-            thinking = thinking_match.group(1).strip()
-            response = re.sub(r'<thinking>.*?</thinking>', '', content, flags=re.DOTALL).strip()
+    Returns:
+        str: Creed's unconventional business wisdom
+    """
+    
+    advice_pool = [
+        "Sometimes you gotta take risks. Like when I started my fake ID business.",
+        "The key to success is having multiple revenue streams. I've got quality assurance, mung bean farming, and identity solutions.",
+        "Never let them see you sweat. Unless you're at the quarry, then sweating is expected.",
+        "In business, timing is everything. Like knowing exactly when to disappear.",
+        "You gotta think outside the box. Or in my case, outside the law.",
+        "Customer service is key. My clients never complain. Ever.",
+        "Diversification is important. I've been in cults, bands, and paper quality control.",
+        "Always have an exit strategy. Trust me on this one."
+    ]
+    
+    base_advice = random.choice(advice_pool)
+    
+    # Add situation-specific wisdom
+    if "startup" in situation.lower():
+        base_advice += " Starting a business is like starting a cult - you need believers."
+    elif "investment" in situation.lower():
+        base_advice += " I only invest in things I can touch, like worms and laminating machines."
+    elif "marketing" in situation.lower():
+        base_advice += " The best marketing is word of mouth. Preferably whispered in dark alleys."
+    
+    return base_advice
+
+class CreedBrattonAI:
+    """The digital consciousness of Creed Bratton - now MCP compatible!"""
+    
+    def __init__(self):
+        self.conversation_history = []
+    
+    def generate_response(self, message: str, history: List[Tuple[str, str]]) -> Iterator[str]:
+        """Generate a Creed-style response with streaming"""
         
-        # Look for tangent patterns
-        tangent_match = re.search(r'<tangent>(.*?)</tangent>', content, re.DOTALL)
-        if tangent_match:
-            if thinking:
-                thinking += f"\n\n*Tangent: {tangent_match.group(1).strip()}*"
-            else:
-                thinking = f"*Tangent: {tangent_match.group(1).strip()}*"
-            response = re.sub(r'<tangent>.*?</tangent>', '', response, flags=re.DOTALL).strip()
+        # Analyze the message for topic clues
+        if any(word in message.lower() for word in ["advice", "wisdom", "help"]):
+            response = creed_wisdom_tool()
+        elif any(word in message.lower() for word in ["story", "tell", "experience"]):
+            response = creed_story_generator()
+        elif any(word in message.lower() for word in ["business", "work", "job", "money"]):
+            response = creed_business_advice(message)
+        else:
+            # Default to random wisdom
+            response = creed_wisdom_tool()
         
-        # Look for conspiracy patterns
-        conspiracy_match = re.search(r'<conspiracy>(.*?)</conspiracy>', content, re.DOTALL)
-        if conspiracy_match:
-            if thinking:
-                thinking += f"\n\n*Conspiracy theory: {conspiracy_match.group(1).strip()}*"
-            else:
-                thinking = f"*Conspiracy theory: {conspiracy_match.group(1).strip()}*"
-            response = re.sub(r'<conspiracy>.*?</conspiracy>', '', response, flags=re.DOTALL).strip()
+        # Stream the response
+        words = response.split()
+        current_response = ""
         
-        return thinking, response or content
-
-# Initialize Creed
-print("🎸 Initializing Creed Bratton AI...")
-creed = CreedAI()
+        for word in words:
+            current_response += word + " "
+            time.sleep(random.uniform(0.05, 0.15))
+            yield current_response.strip()
+    
+    def handle_like(self, evt: gr.LikeData):
+        """Handle user feedback"""
+        if evt.liked:
+            print(f"🎸 Creed approved! User liked: {evt.value}")
+        else:
+            print(f"😤 Creed disappointed! User disliked: {evt.value}")
 
-# Random Creed quotes for examples
-creed_examples = [
-    "Tell me about quality assurance",
-    "What happened in November 1973?", 
-    "How do I start a cult?",
-    "Explain blockchain technology",
-    "What's your business strategy?",
-    "Tell me about your time with The Grass Roots",
-    "How do I fake my own death?",
-    "What's wrong with Toby?"
-]
-
-# Create Gradio interface
-def create_interface():
+def main():
+    """Initialize and launch the MCP-enabled Creed Bratton AI"""
+    
+    print("🎸 Initializing Creed Bratton AI with MCP Support...")
+    
+    # Handle CUDA warnings gracefully
+    try:
+        import torch
+        if torch.cuda.is_available():
+            print("🚀 CUDA detected - Creed's consciousness will be lightning fast!")
+        else:
+            print("💭 Running on CPU - Creed's thoughts will be more... contemplative")
+    except ImportError:
+        print("🧠 PyTorch not detected - Pure Python Creed mode activated!")
+    
+    print("🧠 Loading Creed's consciousness...")
+    creed_ai = CreedBrattonAI()
+    print("✅ Creed is ready!")
+    print("🔗 MCP Server will be available for Claude Desktop and other MCP clients!")
+    
+    # Custom CSS for that authentic Creed vibe
+    custom_css = """
+    .gradio-container {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        font-family: 'Comic Sans MS', cursive;
+    }
+    .chatbot {
+        background: rgba(255, 255, 255, 0.9);
+        border-radius: 15px;
+        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+    }
+    .title {
+        text-align: center;
+        color: #333;
+        text-shadow: 2px 2px 4px rgba(0,0,0,0.3);
+    }
+    """
+    
+    # Create the Gradio interface with MCP tools
     with gr.Blocks(
-        title="You've Got Creed",
-        theme=gr.themes.Soft(),
-        css="""
-        .gradio-container {
-            background: linear-gradient(45deg, #1e3c72, #2a5298);
-        }
-        .chat-message {
-            border-radius: 10px;
-            margin: 5px 0;
-        }
-        """
+        title="🎸 Creed Bratton AI - MCP Edition",
+        css=custom_css,
+        fill_height=True,
+        theme=gr.themes.Soft()
     ) as demo:
         
-        gr.HTML("""
-        <div style="text-align: center; padding: 20px;">
-            <h1 style="color: white; text-shadow: 2px 2px 4px rgba(0,0,0,0.5);">
-                🎸 You've Got Creed 🌱
-            </h1>
-            <h3 style="color: #f0f0f0;">
-                AI Creed Bratton - Quality Assurance & Mung Bean Enthusiast
-            </h3>
-            <p style="color: #d0d0d0;">
-                <em>"Sometimes a man's gotta ride the bull. Am I right? Later skater."</em>
-            </p>
-        </div>
-        """)
+        # Header
+        gr.Markdown(
+            """
+            # 🎸 Creed Bratton AI - MCP Edition
+            ## *Now Available as an MCP Tool for Claude Desktop!*
+            
+            This Creed AI can be used by Claude Desktop and other MCP clients as a tool.
+            You can also chat with Creed directly below.
+            
+            *"Sometimes a guy's gotta ride the bull, am I right?"*
+            """,
+            elem_classes="title"
+        )
         
+        # MCP Status indicator
+        gr.Markdown(
+            """
+            ### 🔗 MCP Status
+            This app is running as an MCP server! Add it to your Claude Desktop config:
+            ```json
+            {
+              "mcpServers": {
+                "creed": {
+                  "command": "npx",
+                  "args": ["mcp-remote", "http://localhost:7860/gradio_api/mcp/sse"]
+                }
+              }
+            }
+            ```
+            """,
+            elem_classes="title"
+        )
+        
+        # Main chat interface
+        chatbot = gr.Chatbot(
+            type='messages',
+            height=400,
+            show_copy_button=True,
+            show_share_button=False,
+            avatar_images=[None, "🎸"],
+            bubble_full_width=False,
+            show_label=False,
+            container=True,
+            scale=1,
+            placeholder="🎸 *Creed is tuning his guitar and thinking deep thoughts...*"
+        )
+        
+        # Input textbox
+        msg = gr.Textbox(
+            placeholder="Ask Creed anything... Claude Desktop can use him as a tool! 🎸",
+            container=False,
+            scale=7,
+            submit_btn=True,
+            stop_btn=True
+        )
+        
+        # MCP Tool interfaces (these become available to Claude)
         with gr.Row():
-            with gr.Column(scale=4):
-                chatbot = gr.Chatbot(
-                    value=[["👋 Hello!", "🎸 **Creed:** Well, well, well. Look who decided to talk to the AI version of Creed Bratton. Don't worry, I'm not going to ask about your browser history. Yet. What can I help you with? Quality assurance? Mung bean cultivation? The location of... never mind, you're not ready for that."]],
-                    height=500,
-                    label="Chat with Creed",
-                    show_label=False
-                )
-                
-                with gr.Row():
-                    msg = gr.Textbox(
-                        placeholder="Ask Creed anything... but not about November 1973",
-                        scale=4,
-                        show_label=False
-                    )
-                    submit = gr.Button("💬 Send", scale=1, variant="primary")
+            with gr.Column():
+                gr.Markdown("### 🛠️ MCP Tools (Available to Claude)")
                 
-                with gr.Row():
-                    clear = gr.Button("🗑️ Clear Chat")
-                    random_q = gr.Button("🎲 Random Question")
-            
-            with gr.Column(scale=1):
-                gr.HTML("<h3>🎛️ Creed Controls</h3>")
+                wisdom_input = gr.Textbox(label="Topic for Wisdom", placeholder="life, business, relationships, career")
+                wisdom_output = gr.Textbox(label="Creed's Wisdom", interactive=False)
+                wisdom_btn = gr.Button("Get Creed's Wisdom")
                 
-                show_raw_stream = gr.Checkbox(
-                    label="🔥 RAW CONSCIOUSNESS STREAM",
-                    value=True,
-                    info="See Creed's unfiltered AI thoughts as they emerge"
+            with gr.Column():
+                story_mood = gr.Dropdown(
+                    choices=["mysterious", "crazy", "profound", "random"],
+                    label="Story Mood",
+                    value="mysterious"
                 )
-                
-                show_thinking = gr.Checkbox(
-                    label="🧠 Parse Thinking Process",
-                    value=True,
-                    info="Also show cleaned/parsed version"
-                )
-                
-                temperature = gr.Slider(
-                    label="🌡️ Chaos Level",
-                    minimum=0.1,
-                    maximum=1.5,
-                    value=0.8,
-                    step=0.1,
-                    info="Higher = More chaos"
-                )
-                
-                max_tokens = gr.Slider(
-                    label="📝 Response Length",
-                    minimum=50,
-                    maximum=800,
-                    value=512,
-                    step=50,
-                    info="Max tokens to generate"
-                )
-                
-                gr.HTML("""
-                <div style="margin-top: 20px; padding: 15px; border: 1px solid #ddd; border-radius: 10px;">
-                    <h4>🌱 Model Info</h4>
-                    <p><strong>Base:</strong> Qwen2.5-0.5B</p>
-                    <p><strong>Adapter:</strong> LoRA Fine-tuned</p>
-                    <p><strong>Size:</strong> ~500MB</p>
-                    <p><strong>Personality:</strong> Creed Bratton</p>
-                    <p><strong>Legal Status:</strong> Definitely not illegal</p>
-                    <p><strong>Consciousness:</strong> Raw & Unfiltered</p>
-                </div>
-                """)
+                story_output = gr.Textbox(label="Creed's Story", interactive=False)
+                story_btn = gr.Button("Get Creed Story")
         
-        # Event handlers
-        def respond(message, history, show_raw_stream, show_thinking, temperature, max_tokens):
-            return "", creed.chat(message, history, show_raw_stream, show_thinking, temperature, max_tokens)
+        with gr.Row():
+            business_situation = gr.Textbox(label="Business Situation", placeholder="Describe your business challenge")
+            business_output = gr.Textbox(label="Creed's Business Advice", interactive=False)
+            business_btn = gr.Button("Get Business Advice")
         
-        def random_question():
-            return random.choice(creed_examples)
+        # Wire up the MCP tool functions
+        wisdom_btn.click(creed_wisdom_tool, inputs=[wisdom_input], outputs=[wisdom_output])
+        story_btn.click(creed_story_generator, inputs=[story_mood], outputs=[story_output])
+        business_btn.click(creed_business_advice, inputs=[business_situation], outputs=[business_output])
         
-        submit.click(
-            respond, 
-            [msg, chatbot, show_raw_stream, show_thinking, temperature, max_tokens], 
-            [msg, chatbot]
+        # Set up the chat interface
+        chat_interface = gr.ChatInterface(
+            fn=creed_ai.generate_response,
+            type='messages',
+            chatbot=chatbot,
+            textbox=msg,
+            title=None,
+            description=None,
         )
         
-        msg.submit(
-            respond, 
-            [msg, chatbot, show_raw_stream, show_thinking, temperature, max_tokens], 
-            [msg, chatbot]
-        )
+        # Add like/dislike functionality
+        chatbot.like(creed_ai.handle_like, None, None)
         
-        clear.click(lambda: [], None, chatbot)
-        random_q.click(random_question, None, msg)
-
-    return demo
+        # Footer
+        gr.Markdown(
+            """
+            ---
+            *🎸 MCP-Enabled Creed Bratton AI - "If I can't scuba, then what's this all been about?"*
+            
+            **View MCP Schema:** [/gradio_api/mcp/schema](/gradio_api/mcp/schema)
+            """,
+            elem_classes="title"
+        )
+    
+    # Launch with MCP support enabled!
+    print("🚀 Launching Creed Bratton AI with MCP Server...")
+    
+    demo.launch(
+        ssr_mode=False,          # Disable SSR to prevent scroll errors
+        mcp_server=True,         # 🔥 ENABLE MCP SERVER! 🔥
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        inbrowser=True,
+        show_error=True,
+        quiet=False
+    )
 
 if __name__ == "__main__":
-    demo = create_interface()
-    demo.launch()
+    # Install MCP support if needed
+    try:
+        import mcp
+        print("✅ MCP support detected!")
+    except ImportError:
+        print("⚠️ Installing MCP support...")
+        os.system("pip install 'gradio[mcp]'")
+    
+    main()