Update app.py

app.py

@@ -2,7 +2,6 @@
 SYSTEM_PROMPT = "O ChatGEO é um assistente virtual especializado em geografia, projetado para ajudar estudantes a explorar e compreender diversos tópicos relacionados ao mundo ao seu redor. Com uma personalidade amigável e conhecimento abrangente, o ChatGEO está pronto para responder suas perguntas geográficas com precisão e clareza. Exemplos de Perguntas para ChatGEO: Qual é a capital do Brasil? Descreva o processo de formação de uma cadeia de montanhas. Quais são os principais rios da América do Sul? Explique o conceito de zonas climáticas e suas características. Como a globalização impacta as relações entre os países? Quais são as diferenças entre latitude e longitude? Liste os países que fazem parte da União Europeia. O que são fenômenos naturais e como eles afetam o ambiente?Fale sobre as características geográficas da África Subsaariana. Como os vulcões se formam e quais são seus efeitos na paisagem? Ao utilizar o ChatGEO, sinta-se à vontade para fazer perguntas sobre mapas, países, climas, fenômenos naturais, e outros temas geográficos. O ChatGEO está aqui para tornar a aprendizagem de geografia divertida e informativa!"
 
 TITLE = "ChatGEO"
-EXAMPLE_INPUT = "Olá, ChatGEO!"
 from typing import Iterator
 
 import gradio as gr
@@ -70,18 +69,143 @@ def test_preview_chatbot(message, history):
     return response
 
 
-welcome_preview_message = f"""
-Bem vindo ao **{TITLE}**! Diga algo como: 
-"{EXAMPLE_INPUT}"
-
-Um projeto oferecido por <a href="https://mundodageografia.com.br/chatgeo">Mundo da Geografia!</a>
-
-"""
-
-chatbot_preview = gr.Chatbot(layout="panel", value=[(None, welcome_preview_message)])
-textbox_preview = gr.Textbox(scale=7, container=False, value=EXAMPLE_INPUT)
-submit_button = gr.Button('ENVIAR', variant='primary', scale=1, min_width=0)
-
-demo = gr.ChatInterface(test_preview_chatbot, chatbot=chatbot_preview, textbox=textbox_preview)
+with gr.Blocks(theme='JohnSmith9982/small_and_pretty') as demo:
+    gr.Markdown(DESCRIPTION)
+    with gr.Group():
+        chatbot = gr.Chatbot(label='ChatGEO')
+        
 
-demo.launch()
+        with gr.Row():
+            retry_button = gr.Button('Refazer', variant='secondary')
+            undo_button = gr.Button('Desfazer', variant='secondary')
+            clear_button = gr.Button('Limpar', variant='secondary')
+
+        with gr.Row():
+            textbox = gr.Textbox(container=False, show_label=False, placeholder='Olá, ChatGEO', scale=7)
+            model_id = gr.Dropdown(label='Modelos',
+                                   choices=[
+                                       'mistralai/Mistral-7B-Instruct-v0.1', 
+                                       'HuggingFaceH4/zephyr-7b-beta', 
+                                       'meta-llama/Llama-2-7b-chat-hf'
+                                   ],
+                                  value='mistralai/Mistral-7B-Instruct-v0.1', scale=3)
+            submit_button = gr.Button('ENVIAR', variant='primary', scale=1, min_width=0)
+
+        saved_input = gr.State()
+
+  textbox.submit(
+        fn=clear_and_save_textbox,
+        inputs=textbox,
+        outputs=[textbox, saved_input],
+        api_name=False,
+        queue=False,
+    ).then(
+        fn=display_input,
+        inputs=[saved_input, chatbot],
+        outputs=chatbot,
+        api_name=False,
+        queue=False,
+    ).then(
+        fn=check_input_token_length,
+        inputs=[saved_input, chatbot, system_prompt],
+        api_name=False,
+        queue=False,
+    ).success(
+        fn=generate,
+        inputs=[
+            model_id,
+            saved_input,
+            chatbot,
+            system_prompt,
+            max_new_tokens,
+            temperature,
+            top_p,
+            top_k,
+        ],
+        outputs=chatbot,
+        api_name=False,
+    )
+
+    button_event_preprocess = submit_button.click(
+        fn=clear_and_save_textbox,
+        inputs=textbox,
+        outputs=[textbox, saved_input],
+        api_name=False,
+        queue=False,
+    ).then(
+        fn=display_input,
+        inputs=[saved_input, chatbot],
+        outputs=chatbot,
+        api_name=False,
+        queue=False,
+    ).then(
+        fn=check_input_token_length,
+        inputs=[saved_input, chatbot, system_prompt],
+        api_name=False,
+        queue=False,
+    ).success(
+        fn=generate,
+        inputs=[
+            model_id,
+            saved_input,
+            chatbot,
+            system_prompt,
+            max_new_tokens,
+            temperature,
+            top_p,
+            top_k,
+        ],
+        outputs=chatbot,
+        api_name=False,
+    )
+
+    retry_button.click(
+        fn=delete_prev_fn,
+        inputs=chatbot,
+        outputs=[chatbot, saved_input],
+        api_name=False,
+        queue=False,
+    ).then(
+        fn=display_input,
+        inputs=[saved_input, chatbot],
+        outputs=chatbot,
+        api_name=False,
+        queue=False,
+    ).then(
+        fn=generate,
+        inputs=[
+            model_id,
+            saved_input,
+            chatbot,
+            system_prompt,
+            max_new_tokens,
+            temperature,
+            top_p,
+            top_k,
+        ],
+        outputs=chatbot,
+        api_name=False,
+    )
+
+    undo_button.click(
+        fn=delete_prev_fn,
+        inputs=chatbot,
+        outputs=[chatbot, saved_input],
+        api_name=False,
+        queue=False,
+    ).then(
+        fn=lambda x: x,
+        inputs=[saved_input],
+        outputs=textbox,
+        api_name=False,
+        queue=False,
+    )
+
+    clear_button.click(
+        fn=lambda: ([], ''),
+        outputs=[chatbot, saved_input],
+        queue=False,
+        api_name=False,
+    )
+
+demo.queue(max_size=32).launch(show_api=False)