Update app.py

app.py

@@ -90,6 +90,13 @@ def format_chat_prompt(message: str, chat_history, bot_name: str) -> str:
 
 
 def chat():
+    gr.HTML(TITLE)
+    with gr.Row():
+        with gr.Column():
+            banner = gr.Image("Albert Einstein.jpeg", elem_id="banner-image", show_label=False)
+        with gr.Column():
+            gr.Markdown(INTRO)
+                
     with gr.Row(elem_id="param_container"):
         with gr.Column():
             model, temperature, top_p = parameter_accordion()
@@ -102,7 +109,8 @@ def chat():
                     interactive=True,
                     label="Character",
                 )
-                
+                bot_name.change(fn=lambda value: gr.update(value=f"{value}.jpeg"}), inputs=bot_name, outputs=banner)
+                        
     with gr.Column(elem_id="chat_container"):
         with gr.Row():
             chatbot = gr.Chatbot(elem_id="chatbot")
@@ -193,14 +201,6 @@ def get_demo():
     min-height: 300px;
 }"""
     ) as demo:
-        gr.HTML(TITLE)
-
-        with gr.Row():
-            with gr.Column():
-                gr.Image("home-banner.jpg", elem_id="banner-image", show_label=False)
-            with gr.Column():
-                gr.Markdown(INTRO)
-
         chat()
 
     return demo