Update app.py

app.py

@@ -13,13 +13,12 @@ co = cohere.Client(cohere_api_key)
 
 history = []
 chat = []
-cid = str(uuid.uuid4())
 
 def trigger_example(example):
     chat, updated_history = generate_response(example)
     return chat, updated_history
         
-def generate_response(user_message, history=None):
+def generate_response(user_message, cid, history=None):
     
     if history is None:
         history = []
@@ -41,14 +40,14 @@ def generate_response(user_message, history=None):
             (history[i].strip(), history[i + 1].strip())
             for i in range(0, len(history) - 1, 2)
         ] 
-        yield chat, history
+        yield chat, history, cid
         
-    return chat, history
+    return chat, history, cid
     
 
-def clear_chat():
-    cid = str(uuid.uuid4())
-    return [], []
+def clear_chat(cid):
+    cid = gr.State(str(uuid.uuid4()))
+    return [], [], cid
 
 
 examples = [
@@ -75,7 +74,8 @@ custom_css = """
 """
 
 with gr.Blocks(analytics_enabled=False, css=custom_css) as demo:
-    
+    cid = gr.State(str(uuid.uuid4()))
+
     with gr.Row():
         with gr.Column(scale=1):
             gr.Image("logoplus.png", elem_id="logo-img", show_label=False, show_share_button=False, show_download_button=False)
@@ -104,14 +104,16 @@ with gr.Blocks(analytics_enabled=False, css=custom_css) as demo:
 
                         
         history = gr.State([])
-        cid = str(uuid.uuid4())
-
         
-        user_message.submit(fn=generate_response, inputs=[user_message, history], outputs=[chatbot, history], concurrency_limit=32)
+        user_message.submit(fn=generate_response, inputs=[user_message, cid, history], outputs=[chatbot, history, cid], concurrency_limit=32)
 
-        submit_button.click(fn=generate_response, inputs=[user_message, history], outputs=[chatbot, history], concurrency_limit=32)
-        clear_button.click(fn=clear_chat, inputs=None, outputs=[chatbot, history], concurrency_limit=32)
+        submit_button.click(fn=generate_response, inputs=[user_message, cid, history], outputs=[chatbot, history, cid], concurrency_limit=32)
+        clear_button.click(fn=clear_chat, inputs=[cid], outputs=[chatbot, history, cid], concurrency_limit=32)
 
+        user_message.submit(lambda x: gr.update(value=""), None, [user_message], queue=False)
+        submit_button.click(lambda x: gr.update(value=""), None, [user_message], queue=False)
+        clear_button.click(lambda x: gr.update(value=""), None, [user_message], queue=False)
+        
         with gr.Row():
             gr.Examples(
                 examples=examples,