Update app.py

app.py

@@ -128,35 +128,20 @@ def ask_agent(question: str) -> str:
             return result_str or "No results."
         return str(result)
     except Exception as e:
-        return f"⚠️ Error: {str(e)}"
+        return f" Error: {str(e)}"
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 🏫 Text to SQL Agent")
-
-    with gr.Row():
-        # --- Left column: tables (smaller) ---
-        with gr.Column(scale=1):
-            gr.Markdown("### Students Table")
-            gr.DataFrame(value=students_df, interactive=False, height=200)
-            gr.Markdown("### Subjects Table")
-            gr.DataFrame(value=subjects_df, interactive=False, height=200)
-            gr.Markdown("### Sample Prompts")
-            for prompt in sample_prompts:
-                gr.Markdown(f"- {prompt}")
-
-        # --- Right column: question & response (larger) ---
-        with gr.Column(scale=2):
-            gr.Markdown("### Ask questions about students, subjects, marks, and teachers")
-            question_input = gr.Textbox(
-                label="Your question",
-                placeholder="e.g., Which student scored highest in Math?",
-                lines=2
-            )
-            answer_output = gr.Textbox(
-                label="Agent response",
-                lines=20
-            )
-
+    gr.Markdown("##  Text to SQL Agent")
+    gr.Markdown("### Students Table")
+    gr.DataFrame(value=students_df, interactive=False)
+    gr.Markdown("### Subjects Table")
+    gr.DataFrame(value=subjects_df, interactive=False)
+    gr.Markdown("### Ask questions about students, subjects, marks, and teachers")
+    question_input = gr.Textbox(label="Your question", placeholder="e.g., Which student scored highest in Math?")
+    answer_output = gr.Textbox(label="Agent response")
+    gr.Markdown("### Sample Prompts")
+    for prompt in sample_prompts:
+        gr.Markdown(f"- {prompt}")
     question_input.submit(fn=ask_agent, inputs=question_input, outputs=answer_output)
 
 demo.launch()