Update app.py

app.py

@@ -1,5 +1,3 @@
-# Revised: RAG system with fallback models if Mistral fails
-
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
@@ -172,33 +170,16 @@ def user_query_with_rag(query, qa_chain, chatbot):
         yield history, ""
 
 def demo():
-    with gr.Blocks(title="RAG Analyzer", css="""
-        .chat-message { display: flex; flex-direction: column; }
-        .chat-message .text { background: #f3f3f3; padding: 8px 12px; border-radius: 10px; max-width: 80%; margin: 4px; }
-        .chat-message.user .text { align-self: flex-end; background: #d1e7dd; }
-        .chat-message.assistant .text { align-self: flex-start; background: #e2e3e5; }
-        .gradio-container { max-width: 1000px !important; margin: auto; font-family: 'Segoe UI', sans-serif; }
-        .gr-button { font-weight: bold; background-color: #0d6efd !important; color: white; }
-    """) as app:
-
+    with gr.Blocks(title="RAG Analyzer") as app:
         db_state = gr.State(None)
         chain_state = gr.State(None)
-
-        gr.Markdown("""
-        # 🧠 Customer Review Analyzer
-        Upload a `.txt` file of reviews and chat with a custom AI system based on your data.
-        """)
-
-        with gr.Row():
-            with gr.Column(scale=1):
-                file_input = gr.File(label="📄 Upload Review File (.txt)", type="filepath")
-                status = gr.Textbox(label="Status", interactive=False)
-                process_btn = gr.Button("🚀 Process Reviews", variant="primary")
-
-            with gr.Column(scale=2):
-                chatbot = gr.Chatbot(label="💬 Chat", height=500, show_copy_button=True, render=False)
-                user_input = gr.Textbox(placeholder="Ask about your reviews...", show_label=False)
-                submit_btn = gr.Button("Send", variant="secondary")
+        gr.Markdown("# 🧠 Customer Review Analyzer with Fallback RAG")
+        file_input = gr.File(label="Upload review file (.txt)", type="filepath")
+        status = gr.Textbox(label="Status")
+        chatbot = gr.Chatbot(label="Chatbot", height=400)
+        user_input = gr.Textbox(placeholder="Ask about the reviews...", show_label=False)
+        submit_btn = gr.Button("Send")
+        process_btn = gr.Button("Process Reviews")
 
         process_btn.click(process_and_initialize, inputs=[file_input], outputs=[db_state, chain_state, status])
         submit_btn.click(user_query_with_rag, inputs=[user_input, chain_state, chatbot], outputs=[chatbot, user_input])
@@ -207,4 +188,4 @@ def demo():
     return app
 
 if __name__ == "__main__":
-    demo().launch(server_name="0.0.0.0", server_port=7860, share=False)
+    demo().launch(server_name="0.0.0.0", server_port=7860, share=False)