Update app.py

app.py

@@ -28,16 +28,14 @@ def predict(message, history, api_key):
         stream=True
     )
 
-    partial_message = ""
+    final_message = ""
     for chunk in response:
         if len(chunk['choices'][0]['delta']) != 0:
-            partial_message = partial_message + chunk['choices'][0]['delta']['content']
+            partial_message = chunk['choices'][0]['delta']['content']
+            final_message += partial_message
             yield partial_message
 
-
-
-    
-    return response["choices"][0]["message"]["content"]
+    return final_message
 
 title = "🕊 Chat with Pigeon"
 
@@ -55,7 +53,7 @@ description = \
 
 👥️️ This demo is by **Evgeniy Hristoforu** (**OpenSkyML**).
 
-<h2></h2>
+
 """
 
 
@@ -69,5 +67,4 @@ examples=[
 
 chatbot = gr.Chatbot(label="PigeonChat",avatar_images=('user.png', 'bot.png'), min_width=260, show_share_button=False, show_copy_button=True, likeable=True, bubble_full_width = False)
 
-
-gr.ChatInterface(predict, chatbot = chatbot, title=title, description=description, examples=examples).launch(show_error=False, show_api=False)
+gr.ChatInterface(predict, chatbot = chatbot, title=title, description=description, examples=examples).launch(show_error=False, show_api=False)