Update app.py

app.py

@@ -12,9 +12,8 @@ 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"
 
 SYSTEM_DEFAULT = (
@@ -90,10 +89,8 @@ 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()
@@ -119,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()
 
 # ---------------------------
@@ -178,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:
@@ -193,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})
@@ -208,14 +202,12 @@ 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)
 
     client = get_client(model_choice)
     api_messages = build_api_messages(sid, system_message)
     partial = ""
-
     try:
         for chunk in client.chat_completion(
             messages=api_messages,
@@ -224,7 +216,6 @@ 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]
@@ -240,7 +231,6 @@ 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)
@@ -254,16 +244,12 @@ 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 = (
-                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()
@@ -276,7 +262,6 @@ 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,7 +270,6 @@ 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]
@@ -301,7 +285,6 @@ 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
@@ -318,90 +301,25 @@ 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.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;
-    }
-    </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")
-            session_list = gr.Radio(
-                choices=labels,
-                value=default_selected,
-                label="Your 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")
-            
-            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()
-            
-                # 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)
+            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")
 
-            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
-            )
+            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 Selection")
-            dataset_choice = gr.Dropdown(
-                choices=DATASETS,
-                value=DATASETS[0],
-                label="Select a dataset",
-                interactive=True
-            )
-
-            gr.Markdown("### ⚙️ Generation Settings")
-            system_box = gr.Textbox(
-                value=SYSTEM_DEFAULT,
-                label="System message",
-                lines=5
-            )
+            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.Column(scale=3):
-            chatbot = gr.Chatbot(label="Assistant", height=520, type="messages")
+            chatbot = gr.Chatbot(label="Assistant", height=720, type="messages")
             with gr.Row():
                 user_box = gr.Textbox(placeholder="Describe your CRUD/backend task…", lines=3, scale=5)
             with gr.Row():
@@ -432,4 +350,4 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()