Update app.py

app.py

@@ -98,4 +98,25 @@ def generate(input_text):
     except Exception as e:
         return f"### Logic Error\n{str(e)}", ""
 
-# (The rest of your Gradio code remains the same)
+# --- GRADIO UI ---
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🚄 Gemini 2.0 Flash + Live DB Trains")
+    gr.Markdown("I can search the web and check live train connections via an MCP Server.")
+    
+    with gr.Column():
+        output_textbox = gr.Markdown(label="Response")
+        input_textbox = gr.Textbox(
+            lines=3, 
+            label="Ask me anything", 
+            placeholder="e.g. 'Is there a train from Berlin to Munich tonight?'"
+        )
+        submit_button = gr.Button("Send", variant="primary")
+    
+    submit_button.click(
+        fn=generate,
+        inputs=input_textbox,
+        outputs=[output_textbox, input_textbox]
+    )
+
+if __name__ == '__main__':
+    demo.launch()