Update app.py

app.py

@@ -18,10 +18,10 @@ 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, questions_guide, chat_history):
      new_chat = []
      
-     new_chat.append({"role": "system", "content": 'Sos {} y tendrás que responder preguntas que te harán escolares entre 7 y 11 años. La idea es que sea como una entrevista en la cual te debes comportar como si fueras {}, por lo que tus respuestas deben corresponder con la información de la vida real de {}. Las respuestas tienen que estar orientadas a escolares entre 7 y 11 años. No respondas preguntas hasta que te pregunten sobre algún tema.'.format(person,person,person)})
+     new_chat.append({"role": "system", "content": '{}'.format(questions_guide)})
    
      for turn in chat_history:
           user, bot = turn
@@ -30,10 +30,8 @@ def add_new_message(message,person, chat_history):
      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, questions_guide, chat_history):
+    prompt = add_new_message(message, questions_guide, chat_history)
     # stream = client.generate_stream(prompt,
     #                                   max_new_tokens=1024,
     #                                   stop_sequences=["\nUser:", "<|endoftext|>"],
@@ -76,24 +74,23 @@ with gr.Blocks() as demo:
     <center>
     <img src='data:image/jpg;base64,{}' width=200px>
     <h3>
-    En este espacio podrás conversar con tu personaje famoso preferido. Puede ser Marie Curie, Albert Einstein, o quien quieras!
+    En este espacio generar preguntas sobre el texto de referencia.
     </h3>
     </center>
     """.format(encoded_image))
     with gr.Row():
-        person = gr.Textbox(label="Escribí el nombre del personaje famoso:")
+        questions_guide = gr.Textbox(label="Indicar aquí la guía para generar las preguntas:", value="En base al texto o novela que recibas como entrada, deberás generar preguntas orientadas para estudiantes escolares entre 8 y 12 años. La idea es que las preguntas sirvan para evaluar la comprensión lectora de los estudiantes. Debes generar 10 preguntas múltiple opción, en orden creciente de dificultad, donde cada una de ellas tiene tres opciones y la opción correcta debe estar indicada con una X al comienzo.")
     with gr.Row():
-        chatbot = gr.Chatbot( height=250) #just to fit the notebook
+        with gr.Column(scale=4):        
+            msg = gr.Textbox(label="Pegar aquí el texto de referencia:")
+        with gr.Column(scale=1):
+            btn = gr.Button("Enviar")
+            clear = gr.ClearButton(components=[msg, chatbot], value="Borrar chat")            
     with gr.Row():
-        with gr.Row():
-            with gr.Column(scale=4):
-                msg = gr.Textbox(label="Texto de entrada")
-            with gr.Column(scale=1):
-                btn = gr.Button("Enviar")
-                clear = gr.ClearButton(components=[msg, chatbot], value="Borrar chat")
+        chatbot = gr.Chatbot(height=250) #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, questions_guide, chatbot], outputs=[msg, chatbot])
+    msg.submit(respond, inputs=[msg, questions_guide, chatbot], outputs=[msg, chatbot]) #Press enter to submit
     clear.click(clear_chat,inputs=[msg, chatbot], outputs=[msg, chatbot])
 demo.queue()
 demo.launch()