Update app.py

app.py

@@ -18,13 +18,12 @@ DATASETS = ["The Stack", "CodeXGLUE"]  # Dropdown for dataset selection
 HF_TOKEN = os.getenv("HF_TOKEN")  # Set in your Space's Secrets
 DB_PATH = "history.db"
 
+# Minimal system prompt (short)
 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 "
-    "(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.'"
+    "Backend-focused assistant.\n"
+    "- Focus: DB, APIs, CRUD, auth, migrations.\n"
+    "- Provide backend scaffolds with paths & commands.\n"
+    "- Frontend only if essential for integration."
 )
 
 # ---------------------------
@@ -184,6 +183,7 @@ FRONTEND_KEYWORDS = [
 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)
@@ -314,11 +314,48 @@ if not labels:
 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>
-      button { background-color: #22c55e !important; color: #ffffff !important; border: none !important; }
-      button:hover { background-color: #16a34a !important; }
-      button:focus { outline: 2px solid #166534 !important; outline-offset: 2px; }
+      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>
     """)
 
@@ -327,58 +364,67 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
     with gr.Row():
         with gr.Column(scale=1, min_width=260):
             gr.Markdown("### 📁 Sessions")
-            session_list = gr.Radio(
+            # compact session selector
+            session_list = gr.Dropdown(
                 choices=labels,
                 value=default_selected,
-                label="Your chats",
+                label="Chats",
                 interactive=True
             )
-            # -----------------------
-            # Editable title input
-            # -----------------------
-            edit_title_box = gr.Textbox(label="✏️ Rename Chat", placeholder="Edit selected chat title...")
-            rename_btn = gr.Button("💾 Save Title")
-            
+
+            # 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
             def rename_session_cb(new_title, selected_label):
                 sid = label_to_id(selected_label)
-                if sid and new_title.strip():
+                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)
-            
+
+            # wire rename button
             rename_btn.click(rename_session_cb, inputs=[edit_title_box, session_list], outputs=session_list)
-            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")
+            # wire other small buttons
+            new_btn.click(new_chat_cb, outputs=[session_list, gr.update(value=[]), gr.update(value="")])
+            del_btn.click(delete_chat_cb, inputs=session_list, outputs=[session_list, gr.update(value=[])])
+            refresh_btn.click(refresh_sessions_cb, outputs=session_list)
+
+            gr.Markdown("### 🤖 Model")
             model_choice = gr.Dropdown(
                 choices=list(MODELS.keys()),
                 value=list(MODELS.keys())[0],
-                label="Choose a model",
+                label="Model",
                 interactive=True
             )
 
-            gr.Markdown("### 📚 Dataset Selection")
+            gr.Markdown("### 📚 Dataset")
             dataset_choice = gr.Dropdown(
                 choices=DATASETS,
                 value=DATASETS[0],
-                label="Select a dataset",
+                label="Dataset",
                 interactive=True
             )
 
-            gr.Markdown("### ⚙️ Generation Settings")
+            gr.Markdown("### ⚙️ Settings")
+            # make system box short (min lines)
             system_box = gr.Textbox(
                 value=SYSTEM_DEFAULT,
                 label="System message",
-                lines=5
+                lines=3
             )
             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")
@@ -392,6 +438,7 @@ 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 (preserve your existing wires)
     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])