Update app.py

app.py

@@ -1,6 +1,8 @@
+import os
+import openai
 from transformers import pipeline, Conversation
 import gradio as gr
-
+import json
 from dotenv import load_dotenv
 
 # Load environment variables from the .env file de forma local
@@ -10,50 +12,22 @@ import base64
 with open("Iso_Logotipo_Ceibal.png", "rb") as image_file:
     encoded_image = base64.b64encode(image_file.read()).decode()
 
-# chatbot = pipeline(model="microsoft/DialoGPT-medium")
-# conversation = Conversation("Hi")
-# response = chatbot(conversation)
-# #conversation.mark_processed()
-# #conversation.append_response(response)
-# conversation.add_user_input("How old are you?")
-
-# conversation2 = chatbot(conversation)
-# print(conversation2)
-
-# def respond(text, conversation):
-#     chatbot = pipeline(model="microsoft/DialoGPT-medium")
-
-#     if len(conversation)==0:
-#         conversation = Conversation(text)
-#         conversation = chatbot(conversation)
-#         print(conversation.iter_texts())
-#         # test = []
-#         # for user,text in conversation.iter_texts():
-            
-
-#         return text, conversation.iter_texts()
-#     else:
-#         conversation.add_user_input(text)
-#         conversation = chatbot(conversation)
-#         return text, conversation.iter_texts()
-
-import os
-import openai
 
 openai.api_key = os.environ['OPENAI_API_KEY']
 
 def clear_chat(message, chat_history):
      return "", []
 
-def add_new_message(message,chat_history):
+def add_new_message(message, chat_history):
      new_chat = []
+        
      for turn in chat_history:
           user, bot = turn
           new_chat.append({"role": "user", "content": user})
           new_chat.append({"role": "assistant","content":bot})
      new_chat.append({"role": "user","content":message})
      return new_chat
-    
+
 def respond(message, chat_history):
     prompt = add_new_message(message, chat_history)
     # stream = client.generate_stream(prompt,
@@ -67,13 +41,14 @@ def respond(message, chat_history):
         messages= prompt,
         temperature=0.5,
         max_tokens=1000,
-        stream = True
+        stream=True,
         )#.choices[0].message.content
-    # chat_history.append((message, response))
+    #chat_history.append((message, response))
 
     token_counter = 0 
     partial_words = "" 
 
+    counter=0
     for chunk in response:
         chunk_message = chunk['choices'][0]['delta']
         if(len(chat_history))<1:
@@ -91,7 +66,6 @@ def respond(message, chat_history):
                     chat_history[-1] =([message,partial_words])
         yield "",chat_history
 
-    # return "",chat_history
 
 with gr.Blocks() as demo:
     gr.Markdown("""
@@ -114,11 +88,10 @@ with gr.Blocks() as demo:
             with gr.Column(scale=1):
                 btn = gr.Button("Enviar")
                 clear = gr.ClearButton(components=[msg, chatbot], value="Borrar chat")
-
+   
     btn.click(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
-    msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot]) #Press enter to submit
+    msg.submit(respond, inputs=[msg ,chatbot], outputs=[msg, chatbot]) #Press enter to submit
     clear.click(clear_chat,inputs=[msg, chatbot], outputs=[msg, chatbot])
-
-
-demo.queue(concurrency_count=4)
+    
+demo.queue()
 demo.launch()