Update app.py

app.py

@@ -267,7 +267,12 @@ def conversation(qa_chain, message, history):
     # return gr.update(value=""), new_history, response_sources[0], response_sources[1] 
     return qa_chain, gr.update(value=""), new_history, response_source1, response_source1_page, response_source2, response_source2_page, response_source3, response_source3_page
     
+def download_csv(chat_history):
+    df = pd.DataFrame(chat_history, columns=['User Message', 'Assistant Response', 'Rating'])
+    df.to_csv('conversation_history.csv', index=False)
+    return 'conversation_history.csv'
 
+    
 def upload_file(file_obj):
     list_file_path = []
     for idx, file in enumerate(file_obj):
@@ -325,6 +330,10 @@ def demo():
 
         with gr.Tab("Step 3 - Conversation with chatbot"):
             chatbot = gr.Chatbot(height=300)
+             # Add rating slider
+            with gr.Row():
+                rating_slider = gr.Slider(minimum=1, maximum=5, label="Rate the message:", interactive=True)
+
             with gr.Accordion("Advanced - Document references", open=False):
                 with gr.Row():
                     doc_source1 = gr.Textbox(label="Reference 1", lines=2, container=True, scale=20)
@@ -340,7 +349,10 @@ def demo():
             with gr.Row():
                 submit_btn = gr.Button("Submit")
                 clear_btn = gr.ClearButton([msg, chatbot])
-            
+                
+            with gr.Row():
+                download_btn = gr.Button("Download Conversation History as CSV")
+
         # Preprocessing events
         #upload_btn.upload(upload_file, inputs=[upload_btn], outputs=[document])
         db_btn.click(initialize_database, \
@@ -353,19 +365,23 @@ def demo():
             outputs=[chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page], \
             queue=False)
 
-        # Chatbot events
+        # Conversation events
         msg.submit(conversation, \
-            inputs=[qa_chain, msg, chatbot], \
+            inputs=[qa_chain, msg, chatbot, rating_slider], \
             outputs=[qa_chain, msg, chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page], \
             queue=False)
         submit_btn.click(conversation, \
-            inputs=[qa_chain, msg, chatbot], \
+            inputs=[qa_chain, msg, chatbot, rating_slider], \
             outputs=[qa_chain, msg, chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page], \
             queue=False)
         clear_btn.click(lambda:[None,"",0,"",0,"",0], \
             inputs=None, \
             outputs=[chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page], \
             queue=False)
+
+        # Add download button event
+        download_btn.click(download_csv, inputs=[chatbot], outputs=None)
+        
     demo.queue().launch(debug=True)