Update app.py

app.py

@@ -10,9 +10,8 @@ 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",
-    "The Stack (Multilingual)": "bigcode/the-stack",
-    "XLCoST (Multilingual)": "reddy-lab-code-research/XLCoST",
-    "CoMA (Multilingual)": "Denilah/CoMA"
+    "StarCoder Backend-Focused": "bigcode/starcoderbase",
+    "CodeLlama Backend-Focused": "codellama/CodeLlama-7b-Instruct"
 }
 
 HF_TOKEN = os.getenv("HF_TOKEN")  # Set in your Space's Secrets
@@ -23,7 +22,8 @@ SYSTEM_DEFAULT = (
     "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."
+    "(e.g., Laravel Blade, Django templates). "
+    "If asked for unrelated frontend/UI tasks, respond politely that you are a backend assistant."
 )
 
 # ---------------------------
@@ -96,8 +96,7 @@ def get_messages(session_id: int):
     """, (session_id,))
     rows = cur.fetchall()
     conn.close()
-    msgs = [{"role": role, "content": content} for (role, content) in rows]
-    return msgs
+    return [{"role": role, "content": content} for role, content in rows]
 
 def add_message(session_id: int, role: str, content: str):
     conn = db()
@@ -259,7 +258,7 @@ if not labels:
     labels, _ = list_sessions()
 default_selected = labels[0] if labels else None
 
-with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Backend-Focused CRUD Assistant", theme=gr.themes.Soft()) as demo:
     gr.HTML("""
     <style>
       button {
@@ -277,7 +276,7 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
     </style>
     """)
 
-    gr.Markdown("## 🗄️ LLaMA & Mistral Backend-Focused CRUD Automation — with Persistent History")
+    gr.Markdown("## 🗄️ Backend-Focused CRUD Automation — with Persistent History")
 
     with gr.Row():
         with gr.Column(scale=1, min_width=260):
@@ -301,7 +300,7 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 interactive=True
             )
 
-                        gr.Markdown("### ⚙️ Generation Settings")
+            gr.Markdown("### ⚙️ Generation Settings")
             system_box = gr.Textbox(
                 value=SYSTEM_DEFAULT,
                 label="System message",
@@ -345,5 +344,3 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
 
 if __name__ == "__main__":
     demo.launch()
-
-