Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 import sqlite3
-import time
 from datetime import datetime
 import gradio as gr
 from huggingface_hub import InferenceClient
@@ -11,14 +10,19 @@ from huggingface_hub import InferenceClient
 MODELS = {
     "Meta LLaMA 3.1 (8B Instruct)": "meta-llama/Llama-3.1-8B-Instruct",
     "Mistral 7B Instruct": "mistralai/Mistral-7B-Instruct-v0.3",
+    # Later you can add your fine-tuned backend-focused model here
+    # "Backend-Finetuned Model": "your-username/backend-crud-model"
 }
-HF_TOKEN = os.getenv("HF_TOKEN")  # set this in your Space's Secrets
+
+HF_TOKEN = os.getenv("HF_TOKEN")  # Set in your Space's Secrets
 DB_PATH = "history.db"
 
 SYSTEM_DEFAULT = (
-    "You are a coding assistant. Respond only with clean and complete code "
-    "unless explanation is explicitly requested. Prefer full CRUD scaffolds, "
-    "with files, paths, and commands when asked."
+    "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). Ignore other frontend/UI tasks."
 )
 
 # ---------------------------
@@ -133,7 +137,6 @@ def build_api_messages(session_id: int, system_message: str):
     return msgs
 
 def get_client(model_choice: str):
-    """Return the right InferenceClient for the chosen model."""
     model_id = MODELS.get(model_choice, list(MODELS.values())[0])
     return InferenceClient(model_id, token=HF_TOKEN)
 
@@ -255,8 +258,7 @@ if not labels:
     labels, _ = list_sessions()
 default_selected = labels[0] if labels else None
 
-with gr.Blocks(title="LLaMA/Mistral CRUD Automation (with History)", theme=gr.themes.Soft()) as demo:
-    # --- Updated CSS to make ALL buttons green ---
+with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.themes.Soft()) as demo:
     gr.HTML("""
     <style>
       button {
@@ -274,7 +276,7 @@ with gr.Blocks(title="LLaMA/Mistral CRUD Automation (with History)", theme=gr.th
     </style>
     """)
 
-    gr.Markdown("## 🦙🤖 LLaMA & Mistral CRUD Automation — with Persistent History")
+    gr.Markdown("## 🗄️ LLaMA & Mistral Backend-Focused CRUD Automation — with Persistent History")
 
     with gr.Row():
         with gr.Column(scale=1, min_width=260):
@@ -311,12 +313,11 @@ with gr.Blocks(title="LLaMA/Mistral CRUD Automation (with History)", theme=gr.th
         with gr.Column(scale=3):
             chatbot = gr.Chatbot(label="Assistant", height=520, type="messages")
             with gr.Row():
-                user_box = gr.Textbox(placeholder="Describe your CRUD task…", lines=3, scale=5)
+                user_box = gr.Textbox(placeholder="Describe your CRUD/backend task…", lines=3, scale=5)
             with gr.Row():
                 send_btn = gr.Button("Send ▶️", variant="primary")
                 regen_btn = gr.Button("Regenerate 🔁", variant="secondary")
 
-    # Interactions
     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])