Update app.py

app.py

@@ -4,6 +4,8 @@ from Chatbot.bot import answer_query
 
 
 def chat_fn(message, history):
+    history = history or []
+
     try:
         answer = answer_query(message)
         if not answer:
@@ -12,78 +14,30 @@ def chat_fn(message, history):
         traceback.print_exc()
         answer = "Something went wrong while processing your request."
 
-    history = history or []
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": answer})
 
     return "", history
 
 
-custom_css = """
-body {
-    margin: 0;
-    background: linear-gradient(135deg, #0f172a, #111827);
-}
-
-.gradio-container {
-    height: 100vh !important;
-    display: flex;
-    flex-direction: column;
-    padding: 0 20px;
-}
-
-.header {
-    padding: 12px 0;
-    font-size: 20px;
-    font-weight: 600;
-}
-
-.subheader {
-    font-size: 13px;
-    color: #9ca3af;
-}
-
-.chat-container {
-    flex: 1;
-    display: flex;
-    flex-direction: column;
-    gap: 10px;
-}
-
-footer {visibility: hidden}
-"""
-
-
-with gr.Blocks(title="RailMind AI") as demo:
+with gr.Blocks(
+    theme=gr.themes.Soft(),
+    title="RailMind AI"
+) as demo:
 
     gr.Markdown("""
-        <div class="header">🚆 RailMind AI</div>
-        <div class="subheader">Real-time railway data & verified rules</div>
+    # 🚆 RailMind AI  
+    Smart railway assistance for live data and verified rules
     """)
 
-    with gr.Column(elem_classes="chat-container"):
-        chatbot = gr.Chatbot(
-            height=480,
-            show_label=False
-        )
+    chatbot = gr.Chatbot(height=420)
 
-        with gr.Row():
-            msg = gr.Textbox(
-                placeholder="Ask about live train status, PNR, fares, seat availability, or railway rules...",
-                show_label=False,
-                scale=8
-            )
-            send = gr.Button("Send", scale=1)
-            clear = gr.Button("Clear", scale=1)
+    msg = gr.Textbox(
+        placeholder="Ask about train schedules, PNR status, seat availability, or railway rules…",
+        show_label=False
+    )
 
     msg.submit(chat_fn, [msg, chatbot], [msg, chatbot])
-    send.click(chat_fn, [msg, chatbot], [msg, chatbot])
-    clear.click(lambda: [], None, chatbot)
 
 
-demo.launch(
-    server_name="0.0.0.0",
-    server_port=7860,
-    theme=gr.themes.Soft(primary_hue="blue", secondary_hue="slate"),
-    css=custom_css
-)
+demo.launch(server_name="0.0.0.0", server_port=7860)