Update app.py

app.py

@@ -90,11 +90,20 @@ def chat_double(history1, history2, system_message, max_tokens, temperature, top
     gen1 = chat(openai.api_model1, history1, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty, openai.api_key1, openai.api_base1)
     gen2 = chat(openai.api_model2, history2, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty, openai.api_key2, openai.api_base2)
 
-    # We define a default value that will be used when one of the generators is exhausted
-    default_value = ([["", ""]], [["", ""]], "")
-    
-    for (chatbot1_out, chat_history_state1_out, message1), (chatbot2_out, chat_history_state2_out, message2) in zip_longest(gen1, gen2, fillvalue=default_value):
-        yield chatbot1_out, chatbot2_out, chat_history_state1_out, chat_history_state2_out, ""
+    # Define a default value that will be used when one of the generators is exhausted
+    latest_chatbot1_out, latest_chat_history_state1_out, _ = ([["", ""]], [["", ""]], "")
+    latest_chatbot2_out, latest_chat_history_state2_out, _ = ([["", ""]], [["", ""]], "")
+
+    for out1, out2 in zip_longest(gen1, gen2, fillvalue=None):
+        if out1 is not None:  # None means gen1 is exhausted
+            chatbot1_out, chat_history_state1_out, _ = out1
+            latest_chatbot1_out, latest_chat_history_state1_out = chatbot1_out, chat_history_state1_out
+            
+        if out2 is not None:  # None means gen2 is exhausted
+            chatbot2_out, chat_history_state2_out, _ = out2
+            latest_chatbot2_out, latest_chat_history_state2_out = chatbot2_out, chat_history_state2_out
+            
+        yield latest_chatbot1_out, latest_chatbot2_out, latest_chat_history_state1_out, latest_chat_history_state2_out, ""
 
 
 start_message = ""