Update app.py

app.py

@@ -12,6 +12,7 @@ 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"]
 HF_TOKEN = os.getenv("HF_TOKEN")
 DB_PATH = "history.db"
@@ -20,13 +21,11 @@ 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.'"
+    "Only include frontend if required for framework integration."
 )
 
 # ---------------------------
-# DB Setup
+# DB Setup (unchanged)
 # ---------------------------
 def db():
     conn = sqlite3.connect(DB_PATH)
@@ -89,8 +88,10 @@ def get_messages(session_id: int):
     conn = db()
     cur = conn.cursor()
     cur.execute("""
-        SELECT role, content FROM messages
-        WHERE session_id = ? ORDER BY id ASC
+        SELECT role, content
+        FROM messages
+        WHERE session_id = ?
+        ORDER BY id ASC
     """, (session_id,))
     rows = cur.fetchall()
     conn.close()
@@ -116,11 +117,11 @@ 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()
 
 # ---------------------------
-# Helpers
+# Helpers (unchanged)
 # ---------------------------
 def label_to_id(label: str | None) -> int | None:
     if not label:
@@ -147,7 +148,7 @@ def load_dataset_by_name(name: str):
     return None
 
 # ---------------------------
-# Gradio Callbacks
+# Gradio Callbacks (unchanged)
 # ---------------------------
 def refresh_sessions_cb():
     labels, _ = list_sessions()
@@ -175,7 +176,7 @@ 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:
@@ -208,6 +209,7 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
     client = get_client(model_choice)
     api_messages = build_api_messages(sid, system_message)
     partial = ""
+
     try:
         for chunk in client.chat_completion(
             messages=api_messages,
@@ -231,6 +233,7 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
                 yield (display_msgs, "", selected_label)
 
         add_message(sid, "assistant", partial)
+
     except Exception as e:
         display_msgs[-1]["content"] = f"⚠️ Error: {str(e)}"
         yield (display_msgs, "", selected_label)
@@ -248,8 +251,11 @@ def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p
         conn = db()
         cur = conn.cursor()
         cur.execute("""
-            DELETE FROM messages WHERE id = (
-                SELECT id FROM messages WHERE session_id=? ORDER BY id DESC LIMIT 1
+            DELETE FROM messages
+            WHERE id = (
+                SELECT id FROM messages
+                WHERE session_id=?
+                ORDER BY id DESC LIMIT 1
             )
         """, (sid,))
         conn.commit()
@@ -262,6 +268,7 @@ def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p
 
     client = get_client(model_choice)
     partial = ""
+
     try:
         for chunk in client.chat_completion(
             messages=api_messages,
@@ -285,12 +292,13 @@ def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p
                 yield display_msgs
 
         add_message(sid, "assistant", partial)
+
     except Exception as e:
         display_msgs[-1]["content"] = f"⚠️ Error: {str(e)}"
         yield display_msgs
 
 # ---------------------------
-# App UI
+# App UI - OPTIMIZED LAYOUT
 # ---------------------------
 init_db()
 labels, _ = list_sessions()
@@ -301,31 +309,152 @@ 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:
-    gr.Markdown("## 🗄️ LLaMA & Mistral Backend-Focused CRUD Automation — with Persistent History")
-
-    with gr.Row():
-        with gr.Column(scale=1, min_width=260):
-            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")
+    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;
+        border: none !important;
+    }
+    button:hover {
+        background-color: #16a34a !important;
+    }
+    button:focus {
+        outline: 2px solid #166534 !important;
+        outline-offset: 2px;
+    }
+    </style>
+    """)
 
-            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("## 🗄️ Backend CRUD Assistant")
 
-            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")
+    with gr.Row():
+        # Left sidebar - compact layout
+        with gr.Column(scale=1, min_width=280):
+            # Sessions section
+            gr.Markdown("### 📁 Sessions", elem_classes=["section-header"])
+            session_list = gr.Radio(
+                choices=labels,
+                value=default_selected,
+                label="Your chats",
+                interactive=True,
+                elem_classes=["compact-row"]
+            )
+            
+            # 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)
+            
+            def rename_session_cb(new_title, selected_label):
+                sid = label_to_id(selected_label)
+                if sid 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)
+            
+            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"]
+            )
+            top_p = gr.Slider(
+                0.1, 1.0, value=0.9, step=0.05,
+                label="Top-p",
+                elem_classes=["compact-slider"]
+            )
 
+        # Main chat area
         with gr.Column(scale=3):
-            chatbot = gr.Chatbot(label="Assistant", height=720, type="messages")
+            chatbot = gr.Chatbot(
+                label="Assistant", 
+                height=520, 
+                type="messages",
+                show_copy_button=True
+            )
             with gr.Row():
-                user_box = gr.Textbox(placeholder="Describe your CRUD/backend task…", lines=3, scale=5)
+                user_box = gr.Textbox(
+                    placeholder="Describe your CRUD/backend task…", 
+                    lines=2,
+                    max_lines=3,
+                    scale=5,
+                    container=False
+                )
             with gr.Row():
-                send_btn = gr.Button("Send ▶️", variant="primary")
-                regen_btn = gr.Button("Regenerate 🔁", variant="secondary")
+                send_btn = gr.Button("Send ▶️", variant="primary", size="sm")
+                regen_btn = gr.Button("Regenerate 🔁", variant="secondary", size="sm")
 
+    # 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])
@@ -350,4 +479,4 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()