Update app.py

app.py

@@ -52,28 +52,35 @@ def predict(question):
     answers = print_answers(result)
     return answers
 
+def respond(message, chat_history):
+    if len(message)==0:
+            message="¿Dónde puedo solicitar asilo?"
+    bot_message = predict(message)[0]['answer']
+    chat_history.append((message, bot_message))
+    return "", chat_history
 
-title = "# Chatbot Refugiados (spanish)"
 description= "Our chatbot helps refugees arriving in Spain by providing information on key topics. \n This project is based on the article titled [Desarrollando un chatbot para refugiados: nuestra experiencia en Saturdays.AI](https://medium.com/saturdays-ai/desarrollando-un-chatbot-para-refugiados-nuestra-experiencia-en-saturdays-ai-9bf2551432c9), which outlines the process of building a chatbot for refugees. \n You can find the training script in this [github repo](https://github.com/jsr90/chatbot_refugiados_train)."
 
-with gr.Blocks() as demo:
-    gr.Markdown(title)
-    gr.Markdown(description)
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="Write your question:", value="¿Dónde puedo solicitar asilo?")
-    submit = gr.Button("Submit")
-    clear = gr.Button("Clear")
-
-    def respond(message, chat_history):
-        if len(message)==0:
-                message="¿Dónde puedo solicitar asilo?"
-        bot_message = predict(message)[0]['answer']
-        chat_history.append((message, bot_message))
-        return "", chat_history
+with gr.Blocks(theme="huggingface") as demo:
+    gr.HTML("<h1 style='text-align: center; font-size: xx-large'>Chatbot Refugiados (spanish)</h1>")
+    gr.HTML("<h2 style='text-align: center; font-size: large'>The demo you're about to see is from a project currently in development.</h2>")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            chatbot = gr.Chatbot()
+        with gr.Column(scale=1):
+            with gr.Row():
+                msg = gr.Textbox(label="Write your question:", value="¿Dónde puedo solicitar asilo?")
+            with gr.Row():
+                submit = gr.Button("Submit")
+                clear = gr.Button("Clear")
+            gr.Image("OIG.jpeg")
 
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
     submit.click(respond, [msg, chatbot], [msg, chatbot])
     clear.click(lambda: None, None, chatbot, queue=False)
 
+    gr.Markdown(description)
+
 if __name__ == "__main__":
     demo.launch()