Update app.py

app.py

@@ -13,19 +13,21 @@ MODELS = {
     "Mistral 7B Instruct": "mistralai/Mistral-7B-Instruct-v0.3",
 }
 
-DATASETS = ["The Stack", "CodeXGLUE"]
-HF_TOKEN = os.getenv("HF_TOKEN")
+DATASETS = ["The Stack", "CodeXGLUE"]  # Dropdown for dataset selection
+HF_TOKEN = os.getenv("HF_TOKEN")  # Set in your Space's Secrets
 DB_PATH = "history.db"
 
 SYSTEM_DEFAULT = (
     "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."
+    "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.'"
 )
 
 # ---------------------------
-# DB Setup (unchanged)
+# DB Setup
 # ---------------------------
 def db():
     conn = sqlite3.connect(DB_PATH)
@@ -121,7 +123,7 @@ def update_session_title_if_needed(session_id: int, first_user_text: str):
     conn.close()
 
 # ---------------------------
-# Helpers (unchanged)
+# Helpers
 # ---------------------------
 def label_to_id(label: str | None) -> int | None:
     if not label:
@@ -148,7 +150,7 @@ def load_dataset_by_name(name: str):
     return None
 
 # ---------------------------
-# Gradio Callbacks (unchanged)
+# Gradio Callbacks
 # ---------------------------
 def refresh_sessions_cb():
     labels, _ = list_sessions()
@@ -183,6 +185,7 @@ 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:
@@ -190,12 +193,14 @@ 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})
@@ -203,6 +208,7 @@ 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)
 
@@ -218,6 +224,7 @@ 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]
@@ -247,6 +254,7 @@ 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()
@@ -277,6 +285,7 @@ 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]
@@ -298,7 +307,7 @@ def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p
         yield display_msgs
 
 # ---------------------------
-# App UI - OPTIMIZED LAYOUT
+# App UI
 # ---------------------------
 init_db()
 labels, _ = list_sessions()
@@ -311,19 +320,6 @@ default_selected = labels[0] if labels else None
 with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.themes.Soft()) as demo:
     gr.HTML("""
     <style>
-    .compact-row {
-        margin-bottom: 0.5rem !important;
-    }
-    .compact-slider {
-        margin-bottom: 0.3rem !important;
-    }
-    .small-textbox {
-        margin-bottom: 0.5rem !important;
-    }
-    .section-header {
-        margin-bottom: 0.5rem !important;
-        font-weight: 600;
-    }
     button {
         background-color: #22c55e !important;
         color: #ffffff !important;
@@ -336,39 +332,41 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
         outline: 2px solid #166534 !important;
         outline-offset: 2px;
     }
+    .compact-sliders .gr-form {
+        margin-bottom: 0.5rem !important;
+    }
+    .compact-sliders .gr-slider {
+        margin-bottom: 0.5rem !important;
+    }
+    .compact-sliders .gr-number {
+        margin-bottom: 0.5rem !important;
+    }
+    .system-message-compact .gr-form {
+        margin-bottom: 0.5rem !important;
+    }
+    .system-message-compact .gr-textarea {
+        min-height: 80px !important;
+        max-height: 120px !important;
+    }
     </style>
     """)
 
-    gr.Markdown("## 🗄️ Backend CRUD Assistant")
+    gr.Markdown("## 🗄️ LLaMA & Mistral Backend-Focused CRUD Automation — with Persistent History")
 
-    with gr.Row():
-        # Left sidebar - compact layout
-        with gr.Column(scale=1, min_width=280):
-            # Sessions section
-            gr.Markdown("### 📁 Sessions", elem_classes=["section-header"])
+    with gr.Row(equal_height=True):
+        with gr.Column(scale=1, min_width=260):
+            gr.Markdown("### 📁 Sessions")
             session_list = gr.Radio(
                 choices=labels,
                 value=default_selected,
                 label="Your chats",
-                interactive=True,
-                elem_classes=["compact-row"]
+                interactive=True
             )
-            
-            # Session management buttons in a tight grid
-            with gr.Row(elem_classes=["compact-row"]):
-                new_btn = gr.Button("➕ New", size="sm", variant="primary")
-                del_btn = gr.Button("🗑️ Delete", size="sm", variant="stop")
-                refresh_btn = gr.Button("🔄", size="sm", variant="secondary")
-            
-            # Rename section - more compact
-            with gr.Row(elem_classes=["compact-row"]):
-                edit_title_box = gr.Textbox(
-                    label="✏️ Rename", 
-                    placeholder="New title...",
-                    scale=3,
-                    elem_classes=["small-textbox"]
-                )
-                rename_btn = gr.Button("💾", size="sm", scale=1)
+            # -----------------------
+            # Editable title input
+            # -----------------------
+            edit_title_box = gr.Textbox(label="✏️ Rename Chat", placeholder="Edit selected chat title...")
+            rename_btn = gr.Button("💾 Save Title")
             
             def rename_session_cb(new_title, selected_label):
                 sid = label_to_id(selected_label)
@@ -378,83 +376,57 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                     cur.execute("UPDATE sessions SET title=? WHERE id=?", (new_title.strip(), sid))
                     conn.commit()
                     conn.close()
+            
+                # Refresh the session list and keep the same one selected
                 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)
             
+            # Connect button to callback
             rename_btn.click(rename_session_cb, inputs=[edit_title_box, session_list], outputs=session_list)
 
-            # Model and Dataset in a compact row
-            with gr.Row(elem_classes=["compact-row"]):
-                with gr.Column(scale=1):
-                    gr.Markdown("**🤖 Model**")
-                    model_choice = gr.Dropdown(
-                        choices=list(MODELS.keys()),
-                        value=list(MODELS.keys())[0],
-                        label="",
-                        interactive=True,
-                        container=False
-                    )
-                with gr.Column(scale=1):
-                    gr.Markdown("**📚 Dataset**")
-                    dataset_choice = gr.Dropdown(
-                        choices=DATASETS,
-                        value=DATASETS[0],
-                        label="",
-                        interactive=True,
-                        container=False
-                    )
-
-            # Generation Settings - more compact
-            gr.Markdown("### ⚙️ Settings", elem_classes=["section-header"])
-            
-            # Compact system message
-            system_box = gr.Textbox(
-                value=SYSTEM_DEFAULT,
-                label="System prompt",
-                lines=2,
-                max_lines=2,
-                elem_classes=["small-textbox"]
-            )
-            
-            # Sliders in compact layout
-            max_tokens = gr.Slider(
-                256, 2048, value=1200, step=16, 
-                label="Max tokens",
-                elem_classes=["compact-slider"]
-            )
-            temperature = gr.Slider(
-                0.0, 2.0, value=0.25, step=0.05,
-                label="Temperature", 
-                elem_classes=["compact-slider"]
+            with gr.Row():
+                new_btn = gr.Button("➕ New Chat", variant="primary")
+                del_btn = gr.Button("🗑️ Delete", variant="stop")
+                refresh_btn = gr.Button("🔄 Refresh", variant="secondary")
+
+            gr.Markdown("### 🤖 Model Selection")
+            model_choice = gr.Dropdown(
+                choices=list(MODELS.keys()),
+                value=list(MODELS.keys())[0],
+                label="Choose a model",
+                interactive=True
             )
-            top_p = gr.Slider(
-                0.1, 1.0, value=0.9, step=0.05,
-                label="Top-p",
-                elem_classes=["compact-slider"]
+
+            gr.Markdown("### 📚 Dataset Selection")
+            dataset_choice = gr.Dropdown(
+                choices=DATASETS,
+                value=DATASETS[0],
+                label="Select a dataset",
+                interactive=True
             )
 
-        # Main chat area
+            gr.Markdown("### ⚙️ Generation Settings")
+            with gr.Group(elem_classes="system-message-compact"):
+                system_box = gr.Textbox(
+                    value=SYSTEM_DEFAULT,
+                    label="System message",
+                    lines=3
+                )
+            
+            with gr.Group(elem_classes="compact-sliders"):
+                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")
+
         with gr.Column(scale=3):
-            chatbot = gr.Chatbot(
-                label="Assistant", 
-                height=520, 
-                type="messages",
-                show_copy_button=True
-            )
+            chatbot = gr.Chatbot(label="Assistant", height=600, type="messages")
             with gr.Row():
-                user_box = gr.Textbox(
-                    placeholder="Describe your CRUD/backend task…", 
-                    lines=2,
-                    max_lines=3,
-                    scale=5,
-                    container=False
-                )
+                user_box = gr.Textbox(placeholder="Describe your CRUD/backend task…", lines=3, scale=5)
             with gr.Row():
-                send_btn = gr.Button("Send ▶️", variant="primary", size="sm")
-                regen_btn = gr.Button("Regenerate 🔁", variant="secondary", size="sm")
+                send_btn = gr.Button("Send ▶️", variant="primary")
+                regen_btn = gr.Button("Regenerate 🔁", variant="secondary")
 
-    # Event handlers (unchanged)
     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])