Update ai_generate.py

ai_generate.py

@@ -27,12 +27,11 @@ def generate(text, model, api):
             stream=True,
             stop=None,
         )
-        
-        for chunk in completion:
-            print(chunk.choices[0].delta.content or "", end="")
 
-        print(response)
-        return response[0] 
+        response = ""
+        for chunk in completion:
+            response += chunk.choices[0].delta.content or ""
+        return response
     elif model == "OpenAI GPT 3.5":
         client = OpenAI(
             api_key=api,