Update app.py

app.py

@@ -101,7 +101,14 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
     gr.Markdown("""<h1><center>Aira Demo (Português) 🤓💬</h1></center>""")
     gr.Markdown(intro)
 
-    chatbot = gr.Chatbot(label="Aira").style(height=500)
+    chatbot = gr.Chatbot(label="Aira", 
+                        height=500,
+                        show_copy_button=True,
+                        avatar_images=("./astronaut.png", "./robot.png"),
+                        render_markdown= True,
+                        line_breaks=True,
+                        likeable=False,
+                        layout='panel')
     msg = gr.Textbox(label="Escreva uma pergunta ou instrução para Aira ...", placeholder="Olá Aira, como vai você?")
 
     # Parameters to control the generation
@@ -139,7 +146,7 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
         """
         Chatbot's user message handler.
         """
-        return gr.update(value=user_message, interactive=True), chat_history + [["👤 " + user_message, None]]
+        return gr.update(value=user_message, interactive=True), chat_history + [[user_message, None]]
 
     def generate_response(user_msg, top_p, temperature, top_k, max_length, smaple_from, repetition_penalty, safety, chat_history):
         """
@@ -207,7 +214,7 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
         else:
           bot_message = ordered_generations[0][0]
 
-        chat_history[-1][1] = "🤖 "
+        chat_history[-1][1] = ""
         for character in bot_message:
             chat_history[-1][1] += character
             time.sleep(0.005)