Update app.py

app.py

@@ -1,7 +1,5 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
-
-client = InferenceClient("cosmosai471/Luna-v2")
+from lunacode import smart_luna_answer
 
 def respond(
     message,
@@ -11,38 +9,23 @@ def respond(
     temperature,
     top_p,
 ):
-    messages = [{"role": "system", "content": system_message}]
-
-    for user_msg, bot_msg in history:
-        if user_msg:
-            messages.append({"role": "user", "content": user_msg})
-        if bot_msg:
-            messages.append({"role": "assistant", "content": bot_msg})
-
-    messages.append({"role": "user", "content": message})
-
-    response = ""
-    for msg in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = msg.choices[0].delta.content or ""
-        response += token
-        yield response
+    # We won't stream but return full output for simplicity
+    try:
+        result = smart_luna_answer(message, max_tokens=max_tokens, temperature=temperature, top_p=top_p)
+        return result
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 demo = gr.ChatInterface(
     fn=respond,
     additional_inputs=[
-        gr.Textbox(value="You are a helpful and friendly AI.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature"),
+        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"),
+        gr.Slider(minimum=0.1, maximum=1.5, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p"),
     ],
-    title="Luna AI - Chat & Code",
-    description="Chat with Luna. She can generate code with web-enhanced context.",
+    title="Luna AI — Code & Web Enhanced Assistant",
+    description="Ask Luna questions, generate code, and explore smart AI-powered answers with web context!",
     theme="default"
 )