Update app.py

app.py

@@ -3,20 +3,47 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import random
 import warnings
+import time
 
-# Suppress the deprecation warning about chatbot type
+# Suppress warnings
 warnings.filterwarnings("ignore", category=UserWarning, message=".*You have not specified a value for the `type` parameter.*")
 
-# Load model with CPU optimization (since you're running on CPU)
+# ASCII Art Banner
+BANNER = """
+╔═══════════════════════════════════════════════════════════════════════════════╗
+║                                                                               ║
+║     ░█████╗░███████╗██████╗░░█████╗░░█████╗░██╗                              ║
+║     ██╔══██╗██╔════╝██╔══██╗██╔══██╗██╔══██╗██║                              ║
+║     ███████║█████╗░░██████╔╝██║░░██║███████║██║                              ║
+║     ██╔══██║██╔══╝░░██╔══██╗██║░░██║██╔══██║██║                              ║
+║     ██║░░██║███████╗██║░░██║╚█████╔╝██║░░██║██║                              ║
+║     ╚═╝░░╚═╝╚══════╝╚═╝░░╚═╝░╚════╝░╚═╝░░╚═╝╚═╝                              ║
+║                                                                               ║
+║                    🚀 NEURAL INTERFACE v2.0 🚀                               ║
+║                    Powered by Microsoft Phi-2                                ║
+║                    █ B L A C K L I N K  L A B S █                           ║
+╚═══════════════════════════════════════════════════════════════════════════════╝
+"""
+
+# Cool loading animation
+def print_loading_animation():
+    loading_chars = ["⚡", "🔥", "💫", "⭐", "✨", "🌟"]
+    for i in range(3):
+        print(f"\r🤖 Initializing Neural Networks {loading_chars[i % len(loading_chars)]}", end="", flush=True)
+        time.sleep(0.3)
+    print()
+
+print(BANNER)
+print_loading_animation()
+
+# Load model
 device = "cuda" if torch.cuda.is_available() else "cpu"
-print(f"🚀 Loading model on {device}...")
+print(f"🔋 Quantum Processing Unit: {device.upper()}")
 
-# Simplified model loading for compatibility
 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/phi-2",
-    torch_dtype=torch.float32,  # Use float32 for CPU
-    trust_remote_code=True,
-    low_cpu_mem_usage=True
+    torch_dtype=torch.float32,
+    trust_remote_code=True
 )
 
 tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2", trust_remote_code=True)
@@ -26,34 +53,33 @@ tokenizer.pad_token = tokenizer.eos_token
 try:
     with open("training-protocol.aero", "r", encoding="utf-8") as f:
         training_protocol = f.read().strip()
-    print("✅ Training protocol loaded successfully")
+    print("✅ Neural Protocol Matrix: LOADED")
 except FileNotFoundError:
     training_protocol = (
         "You are AeroAI, a helpful, friendly, and slightly humorous educational assistant. "
         "If there is any error with finding the file known as the training protocol, resort to this "
         'and do not speak of the "Remorse protocol".'
     )
-    print("⚠️ ERROR9063: training-protocol.aero not found, using Remorse protocol.")
+    print("⚠️ WARNING: Protocol fallback activated")
 
-# Thinking messages
+# Cool thinking messages
 thinking_messages = [
-    "🤔 Thinking deeply about your question...",
-    "📚 Flipping through my mental textbooks...",
-    "🧮 Running some quick calculations...",
-    "💡 Connecting the dots...",
-    "🔍 Double-checking my facts..."
+    "🧠 Accessing neural pathways...",
+    "⚡ Quantum processing in progress...",
+    "🔮 Analyzing data matrices...",
+    "💭 Interfacing with knowledge base...",
+    "🌐 Connecting synaptic networks...",
+    "🔍 Scanning memory banks...",
+    "⚙️ Calibrating response algorithms...",
+    "🚀 Boosting cognitive engines..."
 ]
 
 def chatbot_response(message, history):
-    """
-    Simple chatbot function that works with current Gradio
-    Returns just the response text, not the full history
-    """
     if not message.strip():
-        return "Please ask me something!"
+        return "🤖 Awaiting neural input..."
     
     try:
-        # Build conversation context with limited history
+        # Build conversation with limited history
         recent_history = history[-5:] if len(history) > 5 else history
         
         conversation = training_protocol + "\n\nConversation:\n"
@@ -61,7 +87,7 @@ def chatbot_response(message, history):
             conversation += f"User: {user_msg}\nAeroAI: {ai_msg}\n"
         conversation += f"User: {message}\nAeroAI:"
         
-        # Tokenize with safe parameters
+        # Tokenization
         inputs = tokenizer(
             conversation,
             return_tensors="pt",
@@ -70,11 +96,11 @@ def chatbot_response(message, history):
             max_length=512
         )
         
-        # Generate response with conservative parameters for CPU
+        # Generate response
         with torch.no_grad():
             outputs = model.generate(
                 **inputs,
-                max_new_tokens=200,  # Keep your 200 tokens for training protocol
+                max_new_tokens=200,
                 min_new_tokens=10,
                 do_sample=True,
                 temperature=0.7,
@@ -82,11 +108,10 @@ def chatbot_response(message, history):
                 repetition_penalty=1.1,
                 pad_token_id=tokenizer.pad_token_id,
                 eos_token_id=tokenizer.eos_token_id,
-                num_beams=1,  # Fastest generation
+                num_beams=1,
                 early_stopping=True
             )
         
-        # Extract response
         response_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
         
         if "AeroAI:" in response_text:
@@ -94,94 +119,230 @@ def chatbot_response(message, history):
         else:
             response = response_text[len(conversation):].strip()
         
-        # Clean up response
         if not response or len(response) < 5:
-            response = "I'm having trouble generating a response. Could you please try again?"
+            response = "🔄 Neural processing error. Reinitializing response matrix..."
         
         return response
         
     except Exception as e:
-        return f"⚠️ Error: {str(e)[:100]}... Please try again."
+        return f"❌ SYSTEM ERROR: {str(e)[:100]}... Attempting recovery..."
+
+# Custom CSS for that cyberpunk look
+custom_css = """
+/* Dark cyberpunk theme */
+.gradio-container {
+    background: linear-gradient(135deg, #0c0c0c 0%, #1a1a2e 50%, #16213e 100%) !important;
+    font-family: 'Courier New', monospace !important;
+}
+
+/* Header styling */
+.markdown h1 {
+    background: linear-gradient(45deg, #00ff41, #0077ff, #ff0080);
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+    text-align: center;
+    font-size: 2.5em !important;
+    text-shadow: 0 0 20px rgba(0, 255, 65, 0.5);
+    margin-bottom: 20px !important;
+}
+
+/* Chatbot styling */
+.message-wrap {
+    background: rgba(0, 255, 65, 0.1) !important;
+    border: 1px solid #00ff41 !important;
+    border-radius: 10px !important;
+    backdrop-filter: blur(10px) !important;
+}
+
+.message.user {
+    background: linear-gradient(135deg, #ff0080, #0077ff) !important;
+    color: white !important;
+    border-radius: 15px !important;
+}
+
+.message.bot {
+    background: linear-gradient(135deg, #00ff41, #0077ff) !important;
+    color: #000 !important;
+    border-radius: 15px !important;
+}
+
+/* Input box styling */
+.input-component {
+    background: rgba(0, 0, 0, 0.8) !important;
+    border: 2px solid #00ff41 !important;
+    border-radius: 10px !important;
+    color: #00ff41 !important;
+}
+
+/* Button styling */
+.primary {
+    background: linear-gradient(45deg, #00ff41, #0077ff) !important;
+    border: none !important;
+    border-radius: 25px !important;
+    color: black !important;
+    font-weight: bold !important;
+    text-transform: uppercase !important;
+    box-shadow: 0 0 20px rgba(0, 255, 65, 0.5) !important;
+    transition: all 0.3s ease !important;
+}
+
+.primary:hover {
+    transform: scale(1.05) !important;
+    box-shadow: 0 0 30px rgba(0, 255, 65, 0.8) !important;
+}
+
+.secondary {
+    background: linear-gradient(45deg, #ff0080, #ff4040) !important;
+    border: none !important;
+    border-radius: 25px !important;
+    color: white !important;
+    font-weight: bold !important;
+    text-transform: uppercase !important;
+    box-shadow: 0 0 20px rgba(255, 0, 128, 0.5) !important;
+}
+
+/* Glowing text effects */
+.status-text {
+    color: #00ff41 !important;
+    text-shadow: 0 0 10px #00ff41 !important;
+    font-family: 'Courier New', monospace !important;
+}
+
+/* Matrix rain effect for background */
+body {
+    position: relative;
+    overflow-x: hidden;
+}
+
+body::before {
+    content: '';
+    position: fixed;
+    top: 0;
+    left: 0;
+    width: 100%;
+    height: 100%;
+    background-image: 
+        radial-gradient(circle at 20% 20%, rgba(0, 255, 65, 0.1) 0%, transparent 50%),
+        radial-gradient(circle at 80% 80%, rgba(0, 119, 255, 0.1) 0%, transparent 50%),
+        radial-gradient(circle at 40% 60%, rgba(255, 0, 128, 0.1) 0%, transparent 50%);
+    pointer-events: none;
+    z-index: -1;
+}
+
+/* Animated borders */
+.interface-panel {
+    border: 2px solid transparent;
+    background: linear-gradient(45deg, #0c0c0c, #1a1a2e) padding-box,
+                linear-gradient(45deg, #00ff41, #0077ff, #ff0080) border-box;
+    border-radius: 15px;
+    animation: borderPulse 3s ease-in-out infinite;
+}
+
+@keyframes borderPulse {
+    0%, 100% { border-color: #00ff41; }
+    33% { border-color: #0077ff; }
+    66% { border-color: #ff0080; }
+}
+
+/* Scrollbar styling */
+::-webkit-scrollbar {
+    width: 12px;
+}
+
+::-webkit-scrollbar-track {
+    background: #0c0c0c;
+}
+
+::-webkit-scrollbar-thumb {
+    background: linear-gradient(45deg, #00ff41, #0077ff);
+    border-radius: 6px;
+}
+"""
 
-# Create a much simpler Gradio interface to avoid compatibility issues
 def create_interface():
-    with gr.Blocks(title="AeroAI") as demo:
-        gr.Markdown("# 🚀 AeroAI (Phi-2) — By Blacklink Labs")
-        gr.Markdown("Your friendly AI assistant. Ask me anything!")
+    with gr.Blocks(css=custom_css, title="🚀 AeroAI Neural Interface", theme=gr.themes.Base()) as demo:
+        # Epic header
+        gr.Markdown("""
+        # 🚀 ⚡ AERO∆I NEURAL INTERFACE ⚡ 🚀
+        ## 『 QUANTUM COGNITIVE MATRIX v2.0 』
+        ### ▲ BLACKLINK LABORATORIES ▲
+        ---
+        **STATUS:** 🟢 ONLINE | **CORE:** Microsoft Phi-2 | **MODE:** Adaptive Learning
+        """)
+        
+        # System info panel
+        with gr.Row():
+            with gr.Column(scale=2):
+                gr.Markdown(f"""
+                ### 📊 SYSTEM METRICS
+                ```
+                ┌─────────────────────────────────────┐
+                │ QUANTUM PROCESSOR: {device.upper():<15} │
+                │ NEURAL CORES: ACTIVE               │
+                │ MEMORY BANKS: LOADED               │
+                │ PROTOCOL STATUS: SYNCHRONIZED      │
+                │ RESPONSE ENGINE: READY             │
+                └─────────────────────────────────────┘
+                ```
+                """)
+            
+            with gr.Column(scale=2):
+                gr.Markdown("""
+                ### ⚡ CAPABILITIES
+                - 🧠 **Neural Processing**
+                - 💡 **Knowledge Synthesis** 
+                - 🔬 **Problem Analysis**
+                - 🎨 **Creative Generation**
+                - 🤖 **Adaptive Learning**
+                """)
+        
+        # Main chat interface
+        gr.Markdown("### 💬 NEURAL COMMUNICATION CHANNEL")
         
-        # Use the simpler ChatInterface which handles the conversation automatically
         chat_interface = gr.ChatInterface(
             fn=chatbot_response,
-            title="AeroAI Chat",
-            description="Chat with AeroAI powered by Microsoft Phi-2",
             examples=[
-                "What is artificial intelligence?",
-                "Explain quantum physics simply",
-                "Help me with a math problem",
-                "Tell me a joke"
+                "🔬 Explain quantum entanglement",
+                "💻 Help me debug this code", 
+                "🧮 Solve this mathematical equation",
+                "🚀 What's the future of AI?",
+                "🎨 Create a creative story"
             ],
             cache_examples=False,
-            retry_btn=None,
-            undo_btn=None,
-            clear_btn="🗑️ Clear Chat"
-        )
-    
-    return demo
-
-# Alternative simple interface if ChatInterface doesn't work
-def create_simple_interface():
-    def chat_fn(message, history):
-        response = chatbot_response(message, history or [])
-        history = history or []
-        history.append((message, response))
-        return "", history
-    
-    def clear_fn():
-        return "", []
-    
-    with gr.Blocks(title="AeroAI") as demo:
-        gr.Markdown("# 🚀 AeroAI (Phi-2) — By Blacklink Labs")
-        
-        chatbot_ui = gr.Chatbot(
-            label="AeroAI Chat",
-            height=400,
-            type="tuples"  # Explicitly set the type to avoid the warning
-        )
-        
-        msg = gr.Textbox(
-            label="Your message",
-            placeholder="Type your message here...",
-            lines=2
+            retry_btn="🔄 RETRY",
+            undo_btn="↩️ UNDO", 
+            clear_btn="🗑️ PURGE MEMORY",
+            submit_btn="⚡ TRANSMIT",
+            stop_btn="🛑 HALT"
         )
         
-        with gr.Row():
-            send_btn = gr.Button("Send", variant="primary")
-            clear_btn = gr.Button("Clear", variant="secondary")
-        
-        # Event handlers
-        send_btn.click(chat_fn, inputs=[msg, chatbot_ui], outputs=[msg, chatbot_ui])
-        msg.submit(chat_fn, inputs=[msg, chatbot_ui], outputs=[msg, chatbot_ui])
-        clear_btn.click(clear_fn, outputs=[msg, chatbot_ui])
+        # Footer
+        gr.Markdown("""
+        ---
+        <div style="text-align: center;">
+        <span style="color: #00ff41; font-family: monospace;">
+        『 Powered by Advanced Neural Architecture 』<br>
+        ⚡ BLACKLINK LABS © 2024 ⚡<br>
+        🔮 "The Future is Neural" 🔮
+        </span>
+        </div>
+        """)
     
     return demo
 
 if __name__ == "__main__":
-    print("🌟 Starting AeroAI interface...")
+    print("🌟 Initializing AeroAI Neural Interface...")
+    print("🔋 All systems nominal")
+    print("⚡ Ready for neural connection")
     
-    # Try the modern interface first, fall back to simple if needed
-    try:
-        demo = create_interface()
-        print("✅ Using ChatInterface")
-    except Exception as e:
-        print(f"⚠️ ChatInterface failed, using simple interface: {e}")
-        demo = create_simple_interface()
+    demo = create_interface()
     
-    # Launch with safe parameters
     demo.launch(
-        server_name="127.0.0.1",  # Use localhost instead of 0.0.0.0
+        server_name="127.0.0.1",
         server_port=7860,
         share=False,
-        quiet=True,  # Reduce verbose output
+        quiet=True,
         show_error=True,
-        enable_queue=False  # Disable queue to avoid API issues
+        enable_queue=False,
+        favicon_path=None
     )