Streaming Update (NEEDS OPTIMIZATIONS AND FAILSAFES!!!!!!!!!!!!!!!!!!!!!!)

main.py

@@ -26,23 +26,29 @@ class EventHandler(AssistantEventHandler):
   @override
   def on_text_created(self, text) -> None:
     print(f"\nassistant > ", end="", flush=True)
+    
       
   @override
   def on_text_delta(self, delta, snapshot):
     print(delta.value, end="", flush=True)
+    
       
   def on_tool_call_created(self, tool_call):
     print(f"\nassistant > {tool_call.type}\n", flush=True)
+    
   
   def on_tool_call_delta(self, delta, snapshot):
     if delta.type == 'code_interpreter':
       if delta.code_interpreter.input:
         print(delta.code_interpreter.input, end="", flush=True)
+        
       if delta.code_interpreter.outputs:
         print(f"\n\noutput >", flush=True)
+        
         for output in delta.code_interpreter.outputs:
           if output.type == "logs":
             print(f"\n{output.logs}", flush=True)
+            
   
 
 # Setting up OpenAI Client with API Key
@@ -69,7 +75,7 @@ def create_and_stream(transcript):
     },
     event_handler=EventHandler()
   ) as stream:
-    stream.until_done()
+      stream.until_done()
   messages = stream.get_final_messages()
   return messages[0].content[0].text.value