Stream test #2

app.py

@@ -1,24 +1,31 @@
-import openai
 import gradio as gr
+import openai
+
 
 # Getting responses using the OpenAI API
-def response_chatgpt(api_key, message):
-  # OPENAI API KEY
-  openai.api_key = api_key
-  prompt = (f"{message}")
-  response = openai.Completion.create(
-      engine="text-davinci-003",
-      prompt=prompt,
-      max_tokens=1024
-  )
-  # Displaying the answer on the screen
-  result = response["choices"][0]["text"]
-  return result
-      
-# User input
+def response_chatgpt(api_key, message: str):
+    # OPENAI API KEY
+    openai.api_key = api_key
+    response = openai.Completion.create(
+        engine="text-davinci-003",
+        prompt=message,
+        max_tokens=1024,
+        stream=False,
+    )
+    # Choosing best conversation
+    completion_text = ''
+    # iterate through the stream of events
+    for event in response:
+        event_text = event['choices'][0]['text']  # extract the text
+        completion_text += event_text  # append the text
+
+    return completion_text
+
+
+# User input and web interface
 chatbot = gr.Interface(
-    fn=response_chatgpt, 
+    fn=response_chatgpt,
     inputs=["text", "text"],
     outputs="text",
 )
-chatbot.launch()   
+chatbot.launch()