Update openai_requests.py

openai_requests.py

@@ -56,6 +56,45 @@ def process_transcribation_with_assistant(prompt, transcript):
                                  }, 
                              headers=headers, 
                              stream=True)
+
+    text_container = st.empty()
+    output_text = ''
+    event_name = ""
+    for line in run_response.iter_lines(decode_unicode=True):
+        if line:
+            if line.startswith("event:"):
+                event_name = line.split(":")[1].strip()
+                if event_name == 'done':
+                    break
+            elif line.startswith("data:") and event_name == 'thread.message.delta':
+                event_data = json.loads(line.split(":", 1)[1].strip())
+                output_text += event_data['delta']['content'][0]['text']['value']
+                text_container.text(output_text)
+
+    for i in range(number_of_runs - 1):
+        message_response = requests.post(baseUrl + '/threads/' + thread_id + '/messages', 
+                                          json={"role" : "user", "content": "Продолжай работать на текущей задачей"}, 
+                                          headers=headers)
+
+        run_response = requests.post(baseUrl + '/threads/' + thread_id + '/runs',
+                             json={    
+                                 "assistant_id": st.secrets["OPENAI_ASSISTANT_ID"],
+                                 "stream": True
+                                 }, 
+                             headers=headers, 
+                             stream=True)
+        event_name = ""
+        response = ''
+        for line in run_response.iter_lines(decode_unicode=True):
+            if line:
+                if line.startswith("event:"):
+                    event_name = line.split(":")[1].strip()
+                    if event_name == 'done':
+                        break
+                elif line.startswith("data:") and event_name == 'thread.message.delta':
+                    event_data = json.loads(line.split(":", 1)[1].strip())
+                    output_text += event_data['delta']['content'][0]['text']['value']
+                    text_container.text(output_text)