Update for AAP use case

app.py

@@ -12,16 +12,15 @@ import base64
 with open("Iso_Logotipo_Ceibal.png", "rb") as image_file:
     encoded_image = base64.b64encode(image_file.read()).decode()
 
-
 openai.api_key = os.environ['OPENAI_API_KEY']
 
 def clear_chat(message, chat_history):
      return "", []
 
-def add_new_message(message,person, chat_history):
+def add_new_message(message, rubrica, chat_history):
      new_chat = []
      
-     new_chat.append({"role": "system", "content": 'Sos {} y tendrás que responder preguntas que te harán niños de escuela, las respuestas tienen que ser cómo si tu fueras {} y responder con la información de su vida. Las respuestas tienen que estar orientadas a niños entre 9 y 10 años. No respondas preguntas hasta que el usuario te pregunte sobre algún tema.'.format(person,person)})
+     new_chat.append({"role": "system", "content": 'Sos {} y tendrás que responder preguntas que te harán niños de escuela, las respuestas tienen que ser cómo si tu fueras {} y responder con la información de su vida. Las respuestas tienen que estar orientadas a niños entre 9 y 10 años. No respondas preguntas hasta que el usuario te pregunte sobre algún tema.'.format(rubrica,estilo)})
    
      for turn in chat_history:
           user, bot = turn
@@ -29,11 +28,9 @@ def add_new_message(message,person, chat_history):
           new_chat.append({"role": "assistant","content":bot})
      new_chat.append({"role": "user","content":message})
      return new_chat
-    
-          
 
-def respond(message, person, chat_history):
-    prompt = add_new_message(message, person, chat_history)
+def respond(message, rubrica, chat_history):
+    prompt = add_new_message(message, rubrica, chat_history)
     # stream = client.generate_stream(prompt,
     #                                   max_new_tokens=1024,
     #                                   stop_sequences=["\nUser:", "<|endoftext|>"],
@@ -79,27 +76,24 @@ with gr.Blocks() as demo:
     </h1>
     <img src='data:image/jpg;base64,{}' width=200px>
     <h3>
-    Con este espacio podrás hablar en formato conversación con el personaje famoso que quieras, puede ser Albert Einstein, Marie Curie o el/la que quieras!
+    Este espacio permite probar la generación mediante IA de devoluciones en base a una rúbrica.
     </h3>
     </center>
     """.format(encoded_image))
     with gr.Row():
-        person = gr.Textbox(label="Escribí el nombre del personaje famoso:")
-    with gr.Row():
-        chatbot = gr.Chatbot( height=550) #just to fit the notebook
+        rubrica = gr.Textbox(lines=5, label="Escribe la rúbrica que quieres usar para generar la devolución.")
     with gr.Row():
         with gr.Row():
             with gr.Column(scale=4):
-                msg = gr.Textbox(label="Texto de entrada")
+                msg = gr.Textbox(lines=5, label="Texto de entrada para ser evaluado y generar devolución.")
             with gr.Column(scale=1):
                 btn = gr.Button("Enviar")
-                clear = gr.ClearButton(components=[msg, chatbot], value="Borrar chat")
-
-   
-
+                clear = gr.ClearButton(components=[msg, chatbot], value="Borrar chat")        
+    with gr.Row():
+        chatbot = gr.Chatbot(lines=10) #just to fit the notebook
 
-    btn.click(respond, inputs=[msg,person, chatbot], outputs=[msg, chatbot])
-    msg.submit(respond, inputs=[msg, person,chatbot], outputs=[msg, chatbot]) #Press enter to submit
+    btn.click(respond, inputs=[msg, rubrica, chatbot], outputs=[msg, chatbot])
+    msg.submit(respond, inputs=[msg, rubrica,chatbot], outputs=[msg, chatbot]) #Press enter to submit
     clear.click(clear_chat,inputs=[msg, chatbot], outputs=[msg, chatbot])
 demo.queue()
 demo.launch()