Update app.py

app.py

@@ -10,7 +10,7 @@ import io
 import re
 import os
 
-# Embedded call center FAQs (fixed formatting: escaped quotes, consistent rows, no trailing newline)
+# Embedded call center FAQs (fixed formatting: escaped quotes, consistent rows)
 csv_data = """question,answer,call_id,agent_id,timestamp,language
 "How do I reset my password?","Go to the login page, click ""Forgot Password,"" and follow the email instructions.",12345,A001,2025-04-01 10:15:23,en
 "What are your pricing plans?","We offer Basic ($10/month), Pro ($50/month), and Enterprise (custom).",12346,A002,2025-04-01 10:17:45,en
@@ -87,7 +87,7 @@ except Exception as e:
 # RAG process
 def rag_process(query, k=2):
     if not query.strip() or len(query) < 5:
-        return "Invalid query. Please enter a valid question.", [], {}
+        return "Invalid query. Please select a question.", [], {}
     
     start_time = time.perf_counter()
     try:
@@ -141,7 +141,7 @@ def plot_metrics(metrics):
     plt.close()
     return 'rag_plot.png'
 
-# Gradio interface
+# Gradio interface with buttons
 def chat_interface(query):
     try:
         response, retrieved_faqs, metrics = rag_process(query)
@@ -165,27 +165,34 @@ def chat_interface(query):
 custom_css = """
 body { background-color: #2a2a2a; color: #e0e0e0; }
 .gr-box { background-color: #3a3a3a; border: 1px solid #4a4a4a; }
-.gr-button { background-color: #1e90ff; color: white; }
+.gr-button { background-color: #1e90ff; color: white; margin: 5px; }
 .gr-button:hover { background-color: #1c86ee; }
 """
 
+# Get unique questions for buttons (after cleanup)
+unique_questions = faq_data['question'].tolist()
+
 with gr.Blocks(css=custom_css) as demo:
     gr.Markdown("# Customer Experience Bot Demo")
-    gr.Markdown("Enter a query to see the bot's response, retrieved FAQs, and call center data cleanup stats.")
+    gr.Markdown("Select a question to see the bot's response, retrieved FAQs, and call center data cleanup stats.")
     
+    # Create buttons for each question
     with gr.Row():
-        query_input = gr.Textbox(label="Your Query", placeholder="e.g., How do I reset my password?")
-        submit_btn = gr.Button("Submit")
+        for question in unique_questions:
+            gr.Button(question).click(
+                fn=chat_interface,
+                inputs=gr.State(value=question),
+                outputs=[
+                    gr.Textbox(label="Bot Response"),
+                    gr.Textbox(label="Retrieved FAQs"),
+                    gr.Textbox(label="Data Cleanup Stats"),
+                    gr.Image(label="RAG Pipeline Metrics")
+                ]
+            )
     
     response_output = gr.Textbox(label="Bot Response")
     faq_output = gr.Textbox(label="Retrieved FAQs")
     cleanup_output = gr.Textbox(label="Data Cleanup Stats")
     plot_output = gr.Image(label="RAG Pipeline Metrics")
-    
-    submit_btn.click(
-        fn=chat_interface,
-        inputs=query_input,
-        outputs=[response_output, faq_output, cleanup_output, plot_output]
-    )
 
 demo.launch()