Update app.py

app.py

@@ -4,10 +4,9 @@ import os
 import faiss
 import numpy as np
 import json
-import random
 from sentence_transformers import SentenceTransformer
 
-# ✅ Load files
+# ✅ Load data
 with open("texts.json", "r", encoding="utf-8") as f:
     texts = json.load(f)
 
@@ -17,22 +16,13 @@ embed_model = SentenceTransformer("all-MiniLM-L6-v2")
 API_KEY = os.environ.get("OPENROUTER_API_KEY")
 MODEL = "qwen/qwen-2.5-coder-32b-instruct:free"
 
-# ✅ Friendly openers
-friendly_openers = [
-    "Hey! Gotcha covered 🚀",
-    "Sure thing! Here’s what I found 🧠",
-    "Alright, let me break it down 👇",
-    "No worries — I’ve got you 😄",
-    "Yup, here's how it works ⚙️",
-    "Let’s dive into it 💡",
-    "Boom! Here's the scoop 💥"
-]
-
+# ✅ Context Search
 def get_context(query, top_k=5):
     query_vec = embed_model.encode([query])
     D, I = index.search(np.array(query_vec), top_k)
     return "\n".join([texts[i] for i in I[0]])
 
+# ✅ Chat handler
 def chat_fn(message, history):
     headers = {
         "Authorization": f"Bearer {API_KEY}",
@@ -41,9 +31,9 @@ def chat_fn(message, history):
 
     context = get_context(message)
 
-    system_prompt = f"""You are Codex Assistant by LogIQ Curve — a helpful, friendly AI assistant. 
-Talk like you're chatting with a teammate: casual, confident, and clear. Use emojis to make things fun and engaging 😎. 
-Avoid robotic replies. NEVER mention the word "context" to the user. Just use this internally:
+    system_prompt = f"""You are Codex Assistant by LogIQ Curve — a friendly and intelligent assistant. 
+Speak casually and confidently, like you're helping a teammate. Add emojis where it makes sense, but don’t sound robotic.
+Do NOT mention the word 'context'. Use the following to guide your answer:
 
 {context}
 """
@@ -64,30 +54,31 @@ Avoid robotic replies. NEVER mention the word "context" to the user. Just use th
     try:
         response = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload)
         response.raise_for_status()
-        ai_reply = response.json()["choices"][0]["message"]["content"]
-        opener = random.choice(friendly_openers)
-        reply = f"{opener}\n\n{ai_reply}"
+        reply = response.json()["choices"][0]["message"]["content"]
     except Exception as e:
         reply = f"❌ Oops! Something went wrong: {e}"
 
     return reply
 
-# ✅ Custom CSS to hide footer + settings
+# ✅ Custom CSS to remove footer/settings only
 custom_css = """
 footer { display: none !important; }
 button[data-testid="settings-button"] { display: none !important; }
 """
 
-# ✅ Gradio interface
+# ✅ Launch Gradio Chat UI
 with gr.Blocks(css=custom_css) as demo:
     chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Type your message...", show_label=False)
+    with gr.Row():
+        msg = gr.Textbox(placeholder="Type your message here...", show_label=False, scale=8)
+        send_btn = gr.Button("Send", scale=1)
 
     def respond(message, history):
         response = chat_fn(message, history)
         history.append((message, response))
         return history, ""
 
+    send_btn.click(respond, [msg, chatbot], [chatbot, msg])
     msg.submit(respond, [msg, chatbot], [chatbot, msg])
 
 demo.launch()