Update app.py

app.py

@@ -97,6 +97,9 @@ def conversation(qa_chain, message, history):
             *source_texts[:5],  # Unpack exactly 5 source texts
             *source_pages[:5])  # Unpack exactly 5 source pages
 
+def clear_conversation():
+    return gr.update(value=""), [], "", "", "", "", "", 0, 0, 0, 0, 0
+
 def demo():
     with gr.Blocks(theme="base") as demo:
         vector_db = gr.State()
@@ -140,7 +143,12 @@ def demo():
 
         submit_btn.click(conversation, inputs=[qa_chain, msg, chatbot], outputs=[qa_chain, msg, chatbot] + doc_sources + source_pages)
         msg.submit(conversation, inputs=[qa_chain, msg, chatbot], outputs=[qa_chain, msg, chatbot] + doc_sources + source_pages)
-        
+        clear_btn.click(
+            clear_conversation,
+            inputs=[],
+            outputs=[chatbot] + doc_sources + source_pages
+        )
+    
     demo.queue().launch(debug=True)
 
 if __name__ == "__main__":