reverting back

app.py

@@ -32,31 +32,6 @@ model_list = {
     }
 }
 
-# ------------------------
-# Prompt Construction
-# ------------------------
-FEW_SHOT = """Q: Show all department names.
-A: SELECT name FROM department;
-
-Q: Count number of students.
-A: SELECT COUNT(*) FROM student;
-"""
-
-def build_prompt(user_question):
-    return f"""You are an expert SQL assistant. Convert the given question into a valid SQL query.
-
-Instructions:
-- Return only the SQL query.
-- Do not include markdown, explanations, or formatting.
-- Follow Spider dataset SQL syntax.
-
-Examples:
-{FEW_SHOT}
-
-Now answer this:
-Q: {user_question}
-A:"""
-
 # ------------------------
 # Inference + Evaluation Logic
 # ------------------------
@@ -68,16 +43,15 @@ def evaluate_all_models(user_input, expected_sql, chat_history):
         client = model_config["client"]
         model_id = model_config["model_id"]
 
-        prompt = build_prompt(user_input)
-        messages = [{"role": "user", "content": prompt}]
-
+        messages = chat_history + [{"role": "user", "content": user_input}]
         try:
             start_time = time.time()
+
             result = client.chat.completions.create(
                 model=model_id,
                 messages=messages
             )
-            model_sql = result.choices[0].message.content.strip()
+            model_sql = result.choices[0].message.content
             latency = int((time.time() - start_time) * 1000)
 
         except Exception as e:
@@ -116,7 +90,7 @@ def get_random_spider_prompt():
 # Gradio UI
 # ------------------------
 with gr.Blocks() as demo:
-    gr.Markdown("## 🧠 Spider Dataset Model Evaluation with Prompt Engineering")
+    gr.Markdown("## 🧠 Spider Dataset Model Evaluation")
 
     prompt_input = gr.Textbox(label="Your Prompt", lines=3, placeholder="Ask your BI question...")
     expected_sql_display = gr.Textbox(label="Expected SQL", lines=2, interactive=False)
@@ -142,6 +116,5 @@ with gr.Blocks() as demo:
         outputs=[chat_display, chat_memory, evaluation_display]
     )
 
-
 # Launch
 demo.launch()