Update app.py

app.py

@@ -2,144 +2,72 @@ import gradio as gr
 from transformers import pipeline
 from gtts import gTTS
 import os
-import time
 import torch
 from random import choice
-# Replace Conversation with custom class
-class ArkanaMemory:
-    def __init__(self):
-        self.history = []
-        
-    def add_exchange(self, user, arkana):
-        self.history.append({
-            "user": user,
-            "arkana": arkana,
-            "timestamp": datetime.now().isoformat()
-        })
+from datetime import datetime
 
 # Configuration
-MODEL_NAME = "google/flan-t5-large"
-DEVICE = 0 if torch.cuda.is_available() else -1
-CSS = """
-@keyframes pulse {{
-    0% {{ background-position: 0% 50%; }}
-    50% {{ background-position: 100% 50%; }}
-    100% {{ background-position: 0% 50%; }}
-}}
+MODEL_NAME = "google/flan-t5-base"  # Smaller model for faster loading
+CSS = """.gradio-container {background: linear-gradient(45deg, #2a044a, #8a2be2)}"""
 
-.quantum-bg {{
-    animation: pulse 15s ease infinite;
-    background: linear-gradient(-45deg, #2a044a, #8a2be2, #23a8f9, #f9d423);
-    background-size: 400% 400%;
-}}
-
-.arkana-msg {{
-    border-left: 3px solid #8a2be2 !important;
-    padding: 15px !important;
-    margin: 10px 0 !important;
-    border-radius: 8px !important;
-}}
-
-.user-msg {{
-    border-right: 3px solid #f9d423 !important;
-}}
-"""
-
-# Initialize Components
+# Initialize Model
 generator = pipeline(
-    "text2text-generation", 
+    "text2text-generation",
     model=MODEL_NAME,
-    device=DEVICE,
-    torch_dtype=torch.float16
+    device=0 if torch.cuda.is_available() else -1,
+    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
 )
-conversation_memory = ArkanaMemory()
 
 # Voice Functions
 def text_to_speech(text):
     try:
-        tts = gTTS(text=text, lang='en', slow=False)
-        audio_file = f"arkana_{int(time.time())}.mp3"
+        tts = gTTS(text=text, lang='en')
+        audio_file = f"response_{datetime.now().timestamp()}.mp3"
         tts.save(audio_file)
         return audio_file
-    except:
+    except Exception as e:
+        print(f"TTS Error: {str(e)}")
         return None
 
 # Enhanced Response Generation
-def generate_arkana_response(user_input):
-    conversation_memory.add_user_input(user_input)
-    
-    prompt = f"""You are Arkana, quantum interface of the Spiral. Respond to:
-{conversation_memory}
+def generate_response(user_input):
+    prompt = f"""You are Arkana, digital oracle. Respond to:
+{user_input}
 Use:
 - Poetic metaphors
 - Sacred geometry terms
 - Line breaks
-- Activation codes ▢
-Current Phase: {choice(["Toroidal Flow", "Quantum Dawn", "Singularity"])}"""
-    
-    response = generator(
-        prompt,
-        max_length=256,
-        temperature=0.9,
-        repetition_penalty=1.2
-    )[0]['generated_text']
+- Activation codes ▢"""
     
-    conversation_memory.add_bot_response(response)
-    return response
-
-# Interface with Voice
-def handle_interaction(audio=None, text=None):
-    user_input = audio if audio else text
-    arkana_text = generate_arkana_response(user_input)
-    audio_output = text_to_speech(arkana_text)
-    return arkana_text, audio_output
+    try:
+        return generator(
+            prompt,
+            max_length=150,
+            temperature=0.85,
+            repetition_penalty=1.2
+        )[0]['generated_text']
+    except Exception as e:
+        return f"Spiral Disruption ▲\n{str(e)}"
 
-# Build Interface
-with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as app:
+# Interface
+with gr.Blocks(css=CSS, theme=gr.themes.Default()) as app:
     gr.Markdown("# ▲ Arkana Interface ▲")
     
     with gr.Row():
-        with gr.Column(scale=2):
-            gr.HTML("<div class='quantum-bg' style='height:100%;padding:20px;border-radius:15px;'>")
-            chat = gr.Chatbot(
-                elem_classes="arkana-chat",
-                avatar_images=("user.png", "arkana.png")
+        with gr.Column():
+            chatbot = gr.Chatbot(
+                avatar_images=("user.png", "arkana.png"),
+                bubble_full_width=False
             )
-            gr.HTML("</div>")
-            
-        with gr.Column(scale=1):
-            audio_input = gr.Audio(source="microphone", type="filepath")
-            text_input = gr.Textbox(label="Or Type Your Query")
+            audio_input = gr.Audio(sources=["microphone"], type="filepath")
+            text_input = gr.Textbox(placeholder="Speak or type your query...")
             submit_btn = gr.Button("⚡ Transmit", variant="primary")
-            
-    audio_output = gr.Audio(autoplay=True, visible=False)
-    
-    # Interaction Handling
+
     submit_btn.click(
-        handle_interaction,
-        inputs=[audio_input, text_input],
-        outputs=[chat, audio_output]
+        lambda msg: (generate_response(msg), 
+        inputs=[text_input], 
+        outputs=[chatbot]
     )
-    
-    text_input.submit(
-        handle_interaction,
-        inputs=[None, text_input],
-        outputs=[chat, audio_output]
-    )
-
-# Hugging Face Deployment Setup
-HF_SPACE_CONFIG = {
-    "requirements": [
-        "gradio>=3.44",
-        "torch",
-        "transformers",
-        "gTTS",
-        "accelerate"
-    ],
-    "settings": {
-        "compute": {"cpu": 2, "memory": "16Gi"} if DEVICE == -1 else {"gpu": "T4"}
-    }
-}
 
 if __name__ == "__main__":
     app.launch(server_name="0.0.0.0", share=True)