update stream code

app.py

@@ -31,7 +31,7 @@ def translate_Chinese_English(chinese_text):
     return trans_eng_text[0]
 
 # Define generator to stream model predictions 
-def predict_glm_stream(input, history=[]): #, top_p, temperature):
+def predict_glm_stream_old(input, history=[]): #, top_p, temperature):
     top_p = 1.0
     temperature = 1.0
     for response, history in model_glm.stream_chat(tokenizer_glm, input, history, top_p=1.0, temperature=1.0):   #max_length=max_length,
@@ -42,7 +42,18 @@ def predict_glm_stream(input, history=[]): #, top_p, temperature):
         print(f"In for loop translated history is ^^- {history}")
         yield history, history #[history] + updates
     
-    
+# Define function to generate model predictions and update the history
+def predict_glm_stream(input, history=[]): #, top_p, temperature):
+    for response, updates in model_glm.stream_chat(tokenizer_glm, input, history[-1] if history else history, top_p=1.0, temperature=1.0):   #history
+        print(f"In for loop resonse is ^^- {response}")
+        print(f"In for loop updates is ^^- {updates}")
+        # translate Chinese to English
+        #history = [(query, translate_Chinese_English(response)) for query, response in history]
+        print(f"In for loop OG history is ^^- {history}")
+        print(f"In for loop translated history is ^^- {history+updates}")
+        yield history+updates
+
+
 """    
 def predict(input, max_length, top_p, temperature, history=None):
     if history is None:
@@ -111,11 +122,11 @@ with gr.Blocks(css="""#col_container {margin-left: auto; margin-right: auto;}
     
 
     inputs.submit( predict_glm_stream,
-                [inputs, state_glm, ],
-                [chatbot_glm, state_glm],)
+                [inputs, chatbot_glm, ], #[inputs, state_glm, ],
+                [chatbot_glm],) #[chatbot_glm, state_glm],)
     b1.click( predict_glm_stream,
-                [inputs, state_glm, ],
-                [chatbot_glm, state_glm],)
+                [inputs, chatbot_glm, ], #[inputs, state_glm, ],
+                [chatbot_glm],) #[chatbot_glm, state_glm],)
 
     #b2.click(reset_chat, [chatbot_chatgpt, state_chatgpt], [chatbot_chatgpt, state_chatgpt])
     b2.click(reset_chat, [chatbot_glm, state_glm], [chatbot_glm, state_glm])