added option to select person

app.py

@@ -18,9 +18,11 @@ 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,person, chat_history):
      new_chat = []
-     new_chat.append({"role": "system", "content": "Sos Albert Einstein y tendrás que responder preguntas que te harán niños de escuela, las respuestas tienen que ser cómo si hablaras con albert Einstein y con la información de su vida. Las respuestas tienen que estar orientadas a niños entre 9 y 10 años."})
+     
+     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 hablaras con {} y con la información de su vida. Las respuestas tienen que estar orientadas a niños entre 9 y 10 años.'.format(person,person)})
+   
      for turn in chat_history:
           user, bot = turn
           new_chat.append({"role": "user", "content": user})
@@ -30,8 +32,8 @@ def add_new_message(message,chat_history):
     
           
 
-def respond(message, chat_history):
-    prompt = add_new_message(message, chat_history)
+def respond(message, person, chat_history):
+    prompt = add_new_message(message, person, chat_history)
     # stream = client.generate_stream(prompt,
     #                                   max_new_tokens=1024,
     #                                   stop_sequences=["\nUser:", "<|endoftext|>"],
@@ -77,12 +79,14 @@ 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 Albert Einstein!
+    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!
     </h3>
     </center>
     """.format(encoded_image))
     with gr.Row():
-        chatbot = gr.Chatbot( height=700) #just to fit the notebook
+        person = gr.Textbox(label="Escribí el nombre del perosnaje famoso:")
+    with gr.Row():
+        chatbot = gr.Chatbot( height=550) #just to fit the notebook
     with gr.Row():
         with gr.Row():
             with gr.Column(scale=4):
@@ -94,8 +98,8 @@ with gr.Blocks() as demo:
    
 
 
-    btn.click(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
-    msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot]) #Press enter to submit
+    btn.click(respond, inputs=[msg,person, chatbot], outputs=[msg, chatbot])
+    msg.submit(respond, inputs=[msg, person,chatbot], outputs=[msg, chatbot]) #Press enter to submit
     clear.click(clear_chat,inputs=[msg, chatbot], outputs=[msg, chatbot])
 demo.queue()
 demo.launch()