Update app.py

app.py

@@ -3,6 +3,7 @@ import re
 import logging
 import gradio as gr
 import openai
+from itertools import zip_longest
 
 print(os.environ)
 openai.api_base1 = os.environ.get("OPENAI_API_BASE1")
@@ -88,9 +89,14 @@ def chat(api_model, history, system_message, max_tokens, temperature, top_p, top
 def chat_double(history1, history2, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty):
     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)
-    for (chatbot1_out, chat_history_state1_out, message1), (chatbot2_out, chat_history_state2_out, message2) in zip(gen1, gen2):
+
+    # 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, ""
 
+
 start_message = ""
 
 CSS ="""