update ui

app.py

@@ -5,6 +5,7 @@ import requests
 import warnings
 import logging
 import os
+from datasets import load_dataset
 import pandas as pd
 from dotenv import load_dotenv
 
@@ -261,29 +262,62 @@ def submit_message(message, history):
     print(df)  # For console output
     return history, "", df
 
-with gr.Blocks() as demo:
-    gr.Markdown("# BlackBird Customer Support Chat")
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="Your message")
-    clear = gr.Button("Clear")
-    df_output = gr.Dataframe(label="Conversation Analysis")
+def load_customers_dataset():
+    dataset = load_dataset("dwb2023/blackbird-customers", split="train")
+    df = pd.DataFrame(dataset)
+    return df
 
-    submit_event = msg.submit(submit_message, [msg, chatbot], [chatbot, msg, df_output]).then(
-        lambda: "", None, msg
-    )
+def load_orders_dataset():
+    dataset = load_dataset("dwb2023/blackbird-orders", split="train")
+    df = pd.DataFrame(dataset)
+    return df
 
-    example_inputs = [
-        "What's the status of my orders? My Customer id is 2837622",
-        "Can you confirm my customer info and order status? My email is [email protected]",
-        "I'd like to cancel an order",
-        "Can you update my email address to [email protected]?",
-    ]
+example_inputs = [
+    "What's the status of my orders? My Customer id is 2837622",
+    "Can you confirm my customer info and order status? My email is [email protected]",
+    "I'd like to cancel an order",
+    "Can you update my email address to [email protected]?",
+]
+
+# Create Gradio App
+app = gr.Blocks(theme="sudeepshouche/minimalist")
+
+with app:
+    with gr.Tab("Chatbot"):
+        gr.Markdown("# BlackBird Customer Support Chat")
+        with gr.Row():
+            with gr.Column():
+                msg = gr.Textbox(label="Your message")
+                submit = gr.Button("Submit", variant="primary")
+                clear = gr.Button("Clear", variant="secondary")
+                examples = gr.Examples(
+                    examples=example_inputs,
+                    inputs=msg
+                )
+            with gr.Column():
+                chatbot = gr.Chatbot()
+
+        df_output = gr.Dataframe(label="Conversation Analysis")
+
+        def handle_submit(message, history):
+            return submit_message(message, history)
+
+        submit_event = msg.submit(handle_submit, [msg, chatbot], [chatbot, msg, df_output]).then(
+            lambda: "", None, msg
+        )
+
+        submit.click(submit_message, [msg, chatbot], [chatbot, msg, df_output], show_progress="full").then(
+            lambda: "", None, msg
+        )
+
+        clear.click(lambda: None, None, chatbot, queue=False)
+
+    with gr.Tab("Customers"):
+        customers_df = gr.Dataframe(load_customers_dataset(), label="Customers Data")
 
-    examples = gr.Examples(
-        examples=example_inputs,
-        inputs=msg
-    )
+    with gr.Tab("Orders"):
+        orders_df = gr.Dataframe(load_orders_dataset(), label="Orders Data")
 
-    clear.click(lambda: None, None, chatbot, queue=False)
 
-demo.launch()
+if __name__ == "__main__":
+    app.launch()

requirements.txt

@@ -1,3 +1,4 @@
 anthropic
 gradio
-requests
+requests
+datasets