Update app.py

app.py

@@ -12,18 +12,17 @@ MODELS = {
     "Meta LLaMA 3.1 (8B Instruct)": "meta-llama/Llama-3.1-8B-Instruct",
     "Mistral 7B Instruct": "mistralai/Mistral-7B-Instruct-v0.3",
 }
-
-DATASETS = ["The Stack", "CodeXGLUE"]  # Dropdown for dataset selection
-
-HF_TOKEN = os.getenv("HF_TOKEN")  # Set in your Space's Secrets
+DATASETS = ["The Stack", "CodeXGLUE"]
+HF_TOKEN = os.getenv("HF_TOKEN")
 DB_PATH = "history.db"
 
-# Minimal system prompt (short)
 SYSTEM_DEFAULT = (
-    "Backend-focused assistant.\n"
-    "- Focus: DB, APIs, CRUD, auth, migrations.\n"
-    "- Provide backend scaffolds with paths & commands.\n"
-    "- Frontend only if essential for integration."
+    "You are a backend-focused coding assistant. "
+    "Always prioritize database, API, authentication, routing, migrations, and CRUD logic. "
+    "Provide full backend code scaffolds with files, paths, and commands. "
+    "Only include frontend if required for framework integration "
+    "(e.g., Laravel Blade, Django templates). "
+    "If user asks for too much frontend, politely say: 'I'm a backend-focused assistant and cannot provide excessive frontend code.'"
 )
 
 # ---------------------------
@@ -91,8 +90,7 @@ def get_messages(session_id: int):
     cur = conn.cursor()
     cur.execute("""
         SELECT role, content FROM messages
-        WHERE session_id = ?
-        ORDER BY id ASC
+        WHERE session_id = ? ORDER BY id ASC
     """, (session_id,))
     rows = cur.fetchall()
     conn.close()
@@ -118,7 +116,7 @@ def update_session_title_if_needed(session_id: int, first_user_text: str):
         title = first_user_text.strip().split("\n")[0]
         title = (title[:50] + "…") if len(title) > 50 else title
         cur.execute("UPDATE sessions SET title=? WHERE id=?", (title or "New chat", session_id))
-        conn.commit()
+    conn.commit()
     conn.close()
 
 # ---------------------------
@@ -177,14 +175,13 @@ def delete_chat_cb(selected_label):
     return gr.update(choices=labels, value=selected), []
 
 FRONTEND_KEYWORDS = [
-    "react", "vue", "angular", "html", "css", "javascript", "tailwind", "recharts", "typescript"
+    "react","vue","angular","html","css","javascript","tailwind","recharts","typescript"
 ]
 
 def is_frontend_request(user_text: str) -> bool:
     text_lower = user_text.lower()
     return any(kw in text_lower for kw in FRONTEND_KEYWORDS)
 
-# --- Fixed send_cb to show user message ---
 def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens, temperature, top_p, model_choice, dataset_choice, *args):
     sid = label_to_id(selected_label)
     if sid is None:
@@ -192,14 +189,12 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
         labels, _ = list_sessions()
         selected_label = next((lbl for lbl in labels if lbl.startswith(f"{sid} ")), None)
 
-    # Save user message
     add_message(sid, "user", user_text)
     update_session_title_if_needed(sid, user_text)
 
     display_msgs = chatbot_msgs[:]
     display_msgs.append({"role": "user", "content": user_text})
 
-    # Check for frontend-heavy request
     if is_frontend_request(user_text):
         apology = "⚠️ I'm a backend-focused assistant and cannot provide frontend code."
         display_msgs.append({"role": "assistant", "content": apology})
@@ -207,7 +202,6 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
         yield (display_msgs, "", selected_label)
         return
 
-    # Normal backend response
     display_msgs.append({"role": "assistant", "content": "…"})
     yield (display_msgs, "", selected_label)
 
@@ -222,10 +216,8 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
             top_p=float(top_p),
             stream=True,
         ):
-            # --- FIX: handle models that send empty chunks or use message instead of delta ---
             if not hasattr(chunk, "choices") or not chunk.choices:
                 continue
-
             choice = chunk.choices[0]
             delta = ""
             if hasattr(choice, "delta") and choice.delta and getattr(choice.delta, "content", None) is not None:
@@ -243,7 +235,6 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
         display_msgs[-1]["content"] = f"⚠️ Error: {str(e)}"
         yield (display_msgs, "", selected_label)
 
-
 def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p, model_choice, dataset_choice):
     sid = label_to_id(selected_label)
     if sid is None:
@@ -253,13 +244,11 @@ def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p
     if not msgs:
         return [], ""
 
-    # Remove the last assistant message if it exists (to regenerate it)
     if msgs and msgs[-1]["role"] == "assistant":
         conn = db()
         cur = conn.cursor()
         cur.execute("""
-            DELETE FROM messages
-            WHERE id = (
+            DELETE FROM messages WHERE id = (
                 SELECT id FROM messages WHERE session_id=? ORDER BY id DESC LIMIT 1
             )
         """, (sid,))
@@ -281,10 +270,8 @@ def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p
             top_p=float(top_p),
             stream=True,
         ):
-            # --- FIX: handle models that send empty chunks or use message instead of delta ---
             if not hasattr(chunk, "choices") or not chunk.choices:
                 continue
-
             choice = chunk.choices[0]
             delta = ""
             if hasattr(choice, "delta") and choice.delta and getattr(choice.delta, "content", None) is not None:
@@ -302,7 +289,6 @@ def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p
         display_msgs[-1]["content"] = f"⚠️ Error: {str(e)}"
         yield display_msgs
 
-
 # ---------------------------
 # App UI
 # ---------------------------
@@ -311,113 +297,23 @@ labels, _ = list_sessions()
 if not labels:
     first_sid = create_session("New chat")
     labels, _ = list_sessions()
+
 default_selected = labels[0] if labels else None
 
 with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.themes.Soft()) as demo:
-    # Prevent page-level scrolling and tighten UI spacing.
-    gr.HTML("""
-    <style>
-      html, body { overflow: hidden; height: 100%; } /* Lock scrolling on page */
-      .gradio-container { height: 100vh; width: 100vw; } /* Fill window */
-
-      /* Button styling (smaller icons) */
-      .gr-button, button {
-          padding: 6px 8px !important;
-          font-size: 14px !important;
-          min-height: 28px !important;
-      }
-
-      /* Make sidebar compact and scrollable */
-      .gr-block.gr-column:nth-of-type(1) { 
-          max-width: 340px; 
-          overflow-y: auto; 
-          padding-right: 8px;
-      }
-
-      /* Compact dropdown / radio list for sessions */
-      .gr-dropdown, .gr-radio {
-          max-height: 420px !important;
-          overflow-y: auto !important;
-      }
-
-      /* Reduce space used by sliders & textareas */
-      .gr-slider, .gr-textbox, .gr-dropdown, .gr-radio {
-          margin-bottom: 6px !important;
-      }
-
-      /* Make the chat area fill more vertical space */
-      .gr-chat { height: 720px !important; }
-
-      /* Reduce overall margins */
-      .container, .gr-box { margin: 0 !important; padding: 4px !important; }
-
-      /* Make small icon buttons visually compact */
-      .icon-btn { padding: 4px 6px !important; min-width: 34px !important; }
-
-      /* Keep the second column scrollable if content overflows */
-      .gr-block.gr-column:nth-of-type(2) { max-height: calc(100vh - 40px); overflow-y: auto; }
-    </style>
-    """)
-
     gr.Markdown("## 🗄️ LLaMA & Mistral Backend-Focused CRUD Automation — with Persistent History")
 
     with gr.Row():
         with gr.Column(scale=1, min_width=260):
-            gr.Markdown("### 📁 Sessions")
-            # compact session selector
-            session_list = gr.Dropdown(
-                choices=labels,
-                value=default_selected,
-                label="Chats",
-                interactive=True
-            )
+            session_list = gr.Radio(choices=labels, value=default_selected, label="Your chats", interactive=True)
+            new_btn = gr.Button("➕ New Chat", variant="primary")
+            del_btn = gr.Button("🗑️ Delete", variant="stop")
+            refresh_btn = gr.Button("🔄 Refresh", variant="secondary")
 
-            # small inline controls (icons)
-            with gr.Row():
-                new_btn = gr.Button("➕", elem_id="new-chat-btn")
-                rename_btn = gr.Button("✏️", elem_id="rename-chat-btn")
-                del_btn = gr.Button("🗑️", elem_id="delete-chat-btn")
-                refresh_btn = gr.Button("🔄", elem_id="refresh-chat-btn")
-
-            # editable title input (kept but small)
-            edit_title_box = gr.Textbox(label="Rename selected chat", placeholder="Type new title...", lines=1)
-
-            # rename callback (kept unchanged)
-            def rename_session_cb(new_title, selected_label):
-                sid = label_to_id(selected_label)
-                if sid and new_title and new_title.strip():
-                    conn = db()
-                    cur = conn.cursor()
-                    cur.execute("UPDATE sessions SET title=? WHERE id=?", (new_title.strip(), sid))
-                    conn.commit()
-                    conn.close()
-                labels, _ = list_sessions()
-                new_selected = next((lbl for lbl in labels if lbl.startswith(f"{sid} ")), None)
-                return gr.update(choices=labels, value=new_selected)
-
-            gr.Markdown("### 🤖 Model")
-            model_choice = gr.Dropdown(
-                choices=list(MODELS.keys()),
-                value=list(MODELS.keys())[0],
-                label="Model",
-                interactive=True
-            )
+            model_choice = gr.Dropdown(choices=list(MODELS.keys()), value=list(MODELS.keys())[0], label="Choose a model", interactive=True)
+            dataset_choice = gr.Dropdown(choices=DATASETS, value=DATASETS[0], label="Select a dataset", interactive=True)
 
-            gr.Markdown("### 📚 Dataset")
-            dataset_choice = gr.Dropdown(
-                choices=DATASETS,
-                value=DATASETS[0],
-                label="Dataset",
-                interactive=True
-            )
-
-            gr.Markdown("### ⚙️ Settings")
-            # make system box short (min lines)
-            system_box = gr.Textbox(
-                value=SYSTEM_DEFAULT,
-                label="System message",
-                lines=3
-            )
+            system_box = gr.Textbox(value=SYSTEM_DEFAULT, label="System message", lines=5)
             max_tokens = gr.Slider(256, 4096, value=1200, step=16, label="Max tokens")
             temperature = gr.Slider(0.0, 2.0, value=0.25, step=0.05, label="Temperature")
             top_p = gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p")
@@ -430,19 +326,11 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 send_btn = gr.Button("Send ▶️", variant="primary")
                 regen_btn = gr.Button("Regenerate 🔁", variant="secondary")
 
-    # Hook up callbacks (moved here so chatbot & user_box exist)
-    # wire rename button (rename callback returns gr.update)
-    rename_btn.click(rename_session_cb, inputs=[edit_title_box, session_list], outputs=[session_list])
-
-    # wire other small buttons - use components in outputs (not gr.update)
-    refresh_btn.click(refresh_sessions_cb, outputs=[session_list])
+    refresh_btn.click(refresh_sessions_cb, outputs=session_list)
     new_btn.click(new_chat_cb, outputs=[session_list, chatbot, user_box])
-    del_btn.click(delete_chat_cb, inputs=[session_list], outputs=[session_list, chatbot])
-
-    # session load/change
-    session_list.change(load_session_cb, inputs=[session_list], outputs=[chatbot])
+    del_btn.click(delete_chat_cb, inputs=session_list, outputs=[session_list, chatbot])
+    session_list.change(load_session_cb, inputs=session_list, outputs=chatbot)
 
-    # main conversation wiring
     send_btn.click(
         send_cb,
         inputs=[user_box, session_list, chatbot, system_box, max_tokens, temperature, top_p, model_choice, dataset_choice],
@@ -458,7 +346,7 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
     regen_btn.click(
         regenerate_cb,
         inputs=[session_list, system_box, max_tokens, temperature, top_p, model_choice, dataset_choice],
-        outputs=[chatbot]
+        outputs=chatbot
     )
 
 if __name__ == "__main__":