Update app.py

app.py

@@ -1,13 +1,13 @@
 import gradio as gr
 from lunacode import smart_luna_answer, text_to_speech_luna
 
-def respond(message, history, system_message, max_tokens):
-    try:
-        result = smart_luna_answer(message, max_tokens=max_tokens)
-        return result
-    except Exception as e:
-        return f"Error: {str(e)}"
+# TEXT CHAT
+def text_chat(user_input, max_tokens, system_message, history):
+    response = smart_luna_answer(user_input, max_tokens=max_tokens)
+    history.append((user_input, response))
+    return history, history
 
+# VOICE CHAT
 def voice_chat(audio, max_tokens):
     import speech_recognition as sr
     recognizer = sr.Recognizer()
@@ -17,39 +17,43 @@ def voice_chat(audio, max_tokens):
         try:
             text = recognizer.recognize_google(audio_data)
         except:
-            return "Could not understand audio", None
+            return "Sorry, could not understand audio", None
 
     response_text = smart_luna_answer(text, max_tokens=max_tokens)
     audio_path = text_to_speech_luna(response_text)
     return response_text, audio_path
 
-# Voice tab
+# INTERFACE 1: CHAT
+chat_interface = gr.Interface(
+    fn=text_chat,
+    inputs=[
+        gr.Textbox(label="Your Message"),
+        gr.Slider(minimum=128, maximum=2048, value=512, step=1, label="Max new tokens"),
+        gr.Textbox(value="You are a helpful AI assistant.", label="System message"),
+        gr.State([])  # chat history
+    ],
+    outputs=[
+        gr.Chatbot(label="Luna Chat"),
+        gr.State([])
+    ],
+    title="Luna Chat Assistant"
+)
+
+# INTERFACE 2: VOICE
 voice_interface = gr.Interface(
     fn=voice_chat,
     inputs=[
-        gr.Audio(type="filepath", label="Speak to Luna"),
+        gr.Audio(source="microphone", type="filepath", label="Speak to Luna"),
         gr.Slider(minimum=128, maximum=2048, value=512, step=1, label="Max new tokens")
     ],
     outputs=[
         gr.Textbox(label="Luna's Response"),
         gr.Audio(label="Luna Speaks")
     ],
-    title="Luna Voice Assistant",
-    description="Speak to Luna and hear her response!",
-)
-
-# Text chat
-chat_interface = gr.ChatInterface(
-    fn=respond,
-    additional_inputs=[
-        gr.Textbox(value="You are a helpful AI assistant.", label="System message"),
-        gr.Slider(minimum=128, maximum=2048, value=512, step=1, label="Max new tokens"),
-    ],
-    title="Luna AI — Code & Web Enhanced Assistant",
-    description="Ask Luna questions, generate code, and explore smart AI-powered answers with web context!",
+    title="Luna Voice Assistant"
 )
 
-# Tabs
+# TABS
 demo = gr.TabbedInterface(
     [chat_interface, voice_interface],
     ["Chat with Luna", "Voice with Luna"]