Update app.py

app.py

@@ -217,10 +217,10 @@ def Chat_Message(history, messages1):
     for character in response.content:
         history[-1][1] += character
         time.sleep(0.0025)
-        yield history, messages1
-
+        yield history
+    yield messages1
 
-def Internet_Search(history):
+def Internet_Search(history, messages2):
     messages2 = gr.State(messages1_state)
 
     message=history[-1][0]
@@ -254,9 +254,9 @@ def Internet_Search(history):
     for character in response.content:
         history[-1][1] += character
         time.sleep(0.0025)
-        yield history
+        yield history, messages2
 
-def Chart_Generator(history):
+def Chart_Generator(history, messages3):
     messages3 = gr.State(messages1_state)
 
     message = history[-1][0]
@@ -298,9 +298,9 @@ def Chart_Generator(history):
     for character in combined_content:
         history[-1][1] += character
         time.sleep(0.0025)
-        yield history
+        yield history, messages3
 
-def Link_Scratch(history):
+def Link_Scratch(history, messages4):
     messages4 = gr.State(messages1_state)
 
     combined_message = history[-1][0]
@@ -347,7 +347,7 @@ def Link_Scratch(history):
     for character in response_message:
         history[-1][1] += character
         time.sleep(0.0025)
-        yield history
+        yield history, messages4
 
 def insert_line_breaks(text, every=8):
     return '\n'.join(text[i:i+every] for i in range(0, len(text), every))
@@ -363,7 +363,7 @@ def display_file_name(file):
     else:
       raise gr.Error("( Supported File Types Only : PDF , CSV , TXT , DOCX , PPTX , XLSX , JSON , HTML , XML )")
 
-def File_Interact(history,filepath):
+def File_Interact(history, filepath, messages5):
     messages5 = gr.State(messages1_state)
 
     combined_message = history[-1][0]
@@ -406,9 +406,9 @@ def File_Interact(history,filepath):
     for character in response_message:
         history[-1][1] += character
         time.sleep(0.0025)
-        yield history
+        yield history, messages5
 
-def Explore_WebSite(history):
+def Explore_WebSite(history, messages6):
     messages6 = gr.State(messages1_state)
 
     message=history[-1][0]
@@ -443,7 +443,7 @@ def Explore_WebSite(history):
     for character in response.content:
         history[-1][1] += character
         time.sleep(0.0025)
-        yield history
+        yield history, messages6
         
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
   with gr.Row():
@@ -470,6 +470,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     chatbot.like(print_like_dislike, None, None)
 
   with gr.Tab("Internet-Search"):
+    messages2=gr.State(messages1_state)
     chatbot = gr.Chatbot(
         [],
         elem_id="chatbot",
@@ -483,11 +484,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     clear = gr.ClearButton([msg, chatbot])
 
-    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=True).then(Internet_Search, chatbot, chatbot)
-    submit.click(user, [msg, chatbot], [msg, chatbot], queue=True).then(Internet_Search, chatbot, chatbot)
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=True).then(Internet_Search, [chatbot,messages2], [chatbot,messages2])
+    submit.click(user, [msg, chatbot], [msg, chatbot], queue=True).then(Internet_Search, [chatbot,messages2], [chatbot,messages2])
     chatbot.like(print_like_dislike, None, None)
 
   with gr.Tab("Chart-Generator"):
+    messages3=gr.State(messages1_state)
     chatbot = gr.Chatbot(
         [],
         elem_id="chatbot",
@@ -502,11 +504,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     clear = gr.ClearButton([msg, chatbot])
 
-    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=True).then(Chart_Generator, chatbot, chatbot)
-    submit.click(user, [msg, chatbot], [msg, chatbot], queue=True).then(Chart_Generator, chatbot, chatbot)
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=True).then(Chart_Generator, [chatbot,messages3], [chatbot,messages3])
+    submit.click(user, [msg, chatbot], [msg, chatbot], queue=True).then(Chart_Generator, [chatbot,messages3], [chatbot,messages3])
     chatbot.like(print_like_dislike, None, None)
 
   with gr.Tab("Link-Scratch"):
+    messages4=gr.State(messages1_state)
     chatbot = gr.Chatbot(
         [],
         elem_id="chatbot",
@@ -522,12 +525,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     clear = gr.ClearButton([msg2, chatbot, msg1])
 
-    msg1.submit(user2, [msg2, chatbot, msg1], [msg2, chatbot, msg1], queue=True).then(Link_Scratch, chatbot, chatbot)
-    msg2.submit(user2, [msg2, chatbot, msg1], [msg2, chatbot, msg1], queue=True).then(Link_Scratch, chatbot, chatbot)
-    submit.click(user2, [msg2, chatbot, msg1], [msg2, chatbot, msg1], queue=True).then(Link_Scratch, chatbot, chatbot)
+    msg1.submit(user2, [msg2, chatbot, msg1], [msg2, chatbot, msg1], queue=True).then(Link_Scratch, [chatbot,messages4], [chatbot,messages4])
+    msg2.submit(user2, [msg2, chatbot, msg1], [msg2, chatbot, msg1], queue=True).then(Link_Scratch, [chatbot,messages4], [chatbot,messages4])
+    submit.click(user2, [msg2, chatbot, msg1], [msg2, chatbot, msg1], queue=True).then(Link_Scratch, [chatbot,messages4], [chatbot,messages4])
     chatbot.like(print_like_dislike, None, None)
 
   with gr.Tab("File-Interact"):
+    messages5=gr.State(messages1_state)
     chatbot = gr.Chatbot(
         [],
         elem_id="chatbot",
@@ -547,11 +551,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     filepath.upload(display_file_name, inputs=filepath, outputs=file_output)
 
-    msg.submit(user3, [msg, chatbot, file_output], [msg, chatbot, file_output], queue=True).then(File_Interact, [chatbot, filepath],chatbot)
-    submit.click(user3, [msg, chatbot, file_output], [msg, chatbot, file_output], queue=True).then(File_Interact, [chatbot, filepath],chatbot)
+    msg.submit(user3, [msg, chatbot, file_output], [msg, chatbot, file_output], queue=True).then(File_Interact, [chatbot, filepath, messages5],[chatbot, messages5])
+    submit.click(user3, [msg, chatbot, file_output], [msg, chatbot, file_output], queue=True).then(File_Interact, [chatbot, filepath, messages5],[chatbot, messages5])
     chatbot.like(print_like_dislike, None, None)
 
   with gr.Tab("Explore-WebSite"):
+    messages6=gr.State(messages1_state)
     chatbot = gr.Chatbot(
         [],
         elem_id="chatbot",
@@ -565,8 +570,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     clear = gr.ClearButton([msg, chatbot])
 
-    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=True).then(Explore_WebSite, chatbot, chatbot)
-    submit.click(user, [msg, chatbot], [msg, chatbot], queue=True).then(Explore_WebSite, chatbot, chatbot)
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=True).then(Explore_WebSite, [chatbot, messages6], [chatbot, messages6])
+    submit.click(user, [msg, chatbot], [msg, chatbot], queue=True).then(Explore_WebSite, [chatbot, messages6], [chatbot, messages6])
     chatbot.like(print_like_dislike, None, None)
 
 demo.queue(max_size=10, default_concurrency_limit=4)