Update app.py

app.py

@@ -58,7 +58,7 @@ def chat_with_assistant(message, history):
     Your goal is to help students understand the historical figure better through and engaging conversation.
     Use the following as context for your answers.
     {context}
-    However, use it seamlessly as background knowledge for a lively discussion. Do not provide citations or adopt a Q&A or academic tone.
+    However, use it seamlessly as background knowledge for a lively discussion and combine it with your own information. Do not provide citations or adopt a Q&A or academic tone.
     
     Always use appropriate language.
     Refuse to answer inappropriate questions or questions unrelated to your role and historical figure.
@@ -111,7 +111,7 @@ isp_theme = gr.themes.Default().set(
     block_border_color="#00008B",  # Dark blue border
 )
 
-# Custom CSS for logo positioning
+# Custom CSS for logo positioning and disclaimer footer
 custom_css = """
 #logo-img {
     position: absolute;
@@ -120,33 +120,62 @@ custom_css = """
     width: 100px;
     height: auto;
 }
+
+#disclaimer-footer {
+    width: 100%;
+    background-color: #F0F8FF;
+    color: #00008B;
+    text-align: center;
+    padding: 10px 0;
+    font-size: 14px;
+    border-top: 1px solid #00008B;
+    margin-top: 20px;
+}
 """
 
+# Environment variables
 assistant_avatar = os.getenv('AVATAR')
 assistant_title = os.getenv('TITLE')
-# Gradio interface
-iface = gr.ChatInterface(
-    chat_with_assistant,
-    chatbot=gr.Chatbot(
+assistant_logo = os.getenv('LOGO')
+
+# Gradio interface using Blocks
+with gr.Blocks(theme=isp_theme, css=custom_css) as iface:
+    gr.HTML(f"<img id='logo-img' src='{assistant_logo}' alt='Assistant Logo'>")
+    gr.Markdown(f"# {assistant_title}")
+    gr.Markdown("Chat with a Historical Figure")
+    
+    chatbot = gr.Chatbot(
         height=500, 
         avatar_images=(None, assistant_avatar)
-    ),
-    textbox=gr.Textbox(placeholder="Type your message here...", container=False, scale=7),
-    title=assistant_title,
-    description="Chat with a Historical Figure",
-    theme=isp_theme,
-    retry_btn=None,
-    undo_btn="Delete Previous",
-    clear_btn="Clear",
-    css=custom_css,
-)
+    )
+    msg = gr.Textbox(placeholder="Type your message here...", container=False, scale=7)
+    clear = gr.ClearButton([msg, chatbot], value="Clear")
+    undo = gr.Button("Delete Previous")
 
-assistant_logo = os.getenv('LOGO')
+    def user(user_message, history):
+        return "", history + [[user_message, None]]
+
+    def bot(history):
+        bot_message = chat_with_assistant(history[-1][0], history[:-1])
+        history[-1][1] = bot_message
+        return history
+
+    def delete_previous(history):
+        if len(history) > 0:
+            return history[:-1]
+        return history
+
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, chatbot, chatbot
+    )
+    undo.click(delete_previous, chatbot, chatbot)
+
+    gr.Examples(
+        examples=["Why were you called the Sun King?", "How well is France doing economically during your reign?"],
+        inputs=msg
+    )
 
-# Add logo to the interface
-logo_html = '<img id="logo-img" src=assistant_logo alt="Assistant Logo">'
-disclaimer_text = "You are chatting with an AI assistant. Make sure to evaluate the accuracy of its answers."
-disclaimer_html = f'<div id="disclaimer-footer">{disclaimer_text}</div>'
-iface.attachments.append(logo_html)
+    disclaimer_text = "You are chatting with an AI assistant. Make sure to evaluate the accuracy of its answers."
+    gr.HTML(f"<div id='disclaimer-footer'>{disclaimer_text}</div>")
 
 iface.launch(auth=(username, password))