Update app.py

app.py

@@ -6,7 +6,7 @@ import numpy as np
 import json
 from sentence_transformers import SentenceTransformer
 
-# ✅ Load data
+# Load RAG content
 with open("texts.json", "r", encoding="utf-8") as f:
     texts = json.load(f)
 
@@ -16,13 +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"
 
-# ✅ Context Search
+# Search relevant context
 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
+# Chat function
 def chat_fn(message, history):
     headers = {
         "Authorization": f"Bearer {API_KEY}",
@@ -31,9 +31,8 @@ def chat_fn(message, history):
 
     context = get_context(message)
 
-    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:
+    system_prompt = f"""You are Codex Assistant by LogIQ Curve — a helpful, friendly AI assistant. 
+Talk like a smart human. Use the context below to answer:
 
 {context}
 """
@@ -56,29 +55,82 @@ Do NOT mention the word 'context'. Use the following to guide your answer:
         response.raise_for_status()
         reply = response.json()["choices"][0]["message"]["content"]
     except Exception as e:
-        reply = f"❌ Oops! Something went wrong: {e}"
+        reply = f"❌ Error: {e}"
 
     return reply
 
-# ✅ Custom CSS to remove footer/settings only
+# Custom CSS for full screen & styling
 custom_css = """
-footer { display: none !important; }
-button[data-testid="settings-button"] { display: none !important; }
+* {
+    font-family: 'Segoe UI', sans-serif;
+}
+footer, button[data-testid="settings-button"] {
+    display: none !important;
+}
+#chat-container {
+    display: flex;
+    flex-direction: column;
+    height: 100vh;
+}
+#chat-window {
+    flex: 1;
+    overflow-y: auto;
+    padding: 16px;
+}
+#input-row {
+    display: flex;
+    padding: 10px;
+    border-top: 1px solid #ddd;
+}
+textarea {
+    flex: 1;
+    resize: none;
+    padding: 10px;
+    font-size: 1rem;
+}
+button {
+    margin-left: 8px;
+}
+.message.user {
+    background-color: #daf0ff;
+    padding: 10px 15px;
+    border-radius: 10px;
+    align-self: flex-end;
+    margin: 5px 0;
+    max-width: 80%;
+}
+.message.ai {
+    background-color: #f0f0f0;
+    padding: 10px 15px;
+    border-radius: 10px;
+    align-self: flex-start;
+    margin: 5px 0;
+    max-width: 80%;
+}
 """
 
-# ✅ Launch Gradio Chat UI
 with gr.Blocks(css=custom_css) as demo:
-    chatbot = gr.Chatbot()
-    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])
+    chatbot_state = gr.State([])
+
+    with gr.Column(elem_id="chat-container"):
+        chatbox = gr.HTML('<div id="chat-window"></div>', elem_id="chat-window")
+        with gr.Row(elem_id="input-row"):
+            msg = gr.Textbox(placeholder="Type your message here...", lines=1, scale=8, show_label=False)
+            send = gr.Button("Send", scale=1)
+
+    def update_chat(message, state):
+        state.append(("user", message))
+        response = chat_fn(message, [(u, a) for i, (u, a) in enumerate(zip(state[::2], state[1::2]))])
+        state.append(("ai", response))
+
+        html = ""
+        for role, content in state:
+            bubble_class = "user" if role == "user" else "ai"
+            html += f'<div class="message {bubble_class}">{content}</div>'
+
+        return html, state, ""
+
+    send.click(update_chat, [msg, chatbot_state], [chatbox, chatbot_state, msg])
+    msg.submit(update_chat, [msg, chatbot_state], [chatbox, chatbot_state, msg])
 
 demo.launch()