Upload chat_functions.py

functions/chat_functions.py

@@ -89,6 +89,24 @@ with gr.Blocks(css=css) as demo:
     def data_options(filename, request: gr.Request):
         print(filename)
         if filename:
+            if "bank_marketing_campaign" in filename:
+                example_questions = [
+                                        ["Describe the dataset"],
+                                        ["What levels of education have the highest and lowest average balance?"],
+                                        ["What job is most and least common for a yes response from the individuals, not counting 'unknown'?"],
+                                    ]
+            elif "online_retail_data" in filename:
+                example_questions = [
+                                        ["Describe the dataset"],
+                                        ["What month had the highest revenue?"],
+                                        ["Is revenue higher in the morning or afternoon?"],
+                                    ]
+            else:
+                example_questions = [
+                                        ["Describe the dataset"],
+                                        ["List the columns in the dataset"],
+                                        ["What could this data be used for?"],
+                                    ]
             parameters = gr.Textbox(visible=False, value=request.session_hash)
             bot = gr.Chatbot(type='messages', label="CSV Chat Window", show_label=True, render=False, visible=True, elem_classes="chatbot")
             chat = gr.ChatInterface(
@@ -97,11 +115,7 @@ with gr.Blocks(css=css) as demo:
                                 chatbot=bot,
                                 title="Chat with your data file",
                                 concurrency_limit=None,
-                                examples=[
-                                    ["Describe the dataset"],
-                                    ["List the columns in the dataset"],
-                                    ["What could this data be used for?"],
-                                ],
+                                examples=example_questions,
                                 additional_inputs=parameters
                                 ) 
             process_upload(filename, request.session_hash)