Update app.py

app.py

@@ -28,10 +28,13 @@ def generate(message, history, temperature=0.7, max_new_tokens=1024, top_p=0.90,
         seed=random.randint(1, 1111111111111111),
     )
 
+    # Gerar o texto
     output = client.text_generation(formatted_prompt, **generate_kwargs)
 
+    # Extrair a resposta apenas até o primeiro ponto final ou outro delimitador
     response = output.split("\n")[0].strip()
 
+    # Limitar a resposta para que o bot não continue respondendo indefinidamente
     if "Usuário:" in response or "ChatGEO:" in response:
         response = response.split("Usuário:")[0].strip()
 
@@ -39,26 +42,70 @@ def generate(message, history, temperature=0.7, max_new_tokens=1024, top_p=0.90,
     return response
 
 mychatbot = gr.Chatbot(
-    avatar_images=["./user.png", "./botgm.png"],
+    avatar_images=["./user.png", "./botz.png"],
     bubble_full_width=False,
     show_label=False,
     show_copy_button=True,
     likeable=True,
 )
 
-# Substituir additional_inputs por uma informação sobre o chatbot
-chatbot_info = gr.Markdown(
+additional_inputs = [
+
+    chatbot_info = gr.Markdown(
     """
     ### Bem-vindo ao ChatGEO!
     O ChatGEO é um assistente especializado em geografia, pronto para responder suas perguntas sobre o mundo ao nosso redor.
     Faça suas perguntas e explore tópicos interessantes com clareza e precisão.
     """
-)
+    ),
+
+    gr.Slider(
+        label="Temperature",
+        value=0.7,
+        minimum=0.0,
+        maximum=1.0,
+        step=0.01,
+        interactive=True,
+        info="Higher values generate more diverse outputs",
+        visible=False  # Oculta o slider
+
+    ),
+    gr.Slider(
+        label="Max new tokens",
+        value=1024,
+        minimum=0,
+        maximum=8000,
+        step=64,
+        interactive=True,
+        info="The maximum number of new tokens",
+        visible=False  # Oculta o slider
+    ),
+    gr.Slider(
+        label="Top-p",
+        value=0.90,
+        minimum=0.0,
+        maximum=1.0,
+        step=0.01,
+        interactive=True,
+        info="Higher values sample more low-probability tokens",
+        visible=False  # Oculta o slider
+    ),
+    gr.Slider(
+        label="Repetition penalty",
+        value=1.0,
+        minimum=0.1,
+        maximum=2.0,
+        step=0.1,
+        interactive=True,
+        info="Penalize repeated tokens",
+        visible=False  # Oculta o slider
+    )
+]
 
 iface = gr.ChatInterface(
     fn=generate, 
     chatbot=mychatbot,
-    additional_inputs=[chatbot_info],  # Exibe a mensagem sobre o chatbot
+    additional_inputs=additional_inputs,
     submit_btn='Enviar',
     retry_btn=None,
     undo_btn=None,
@@ -66,7 +113,7 @@ iface = gr.ChatInterface(
 )
 
 with gr.Blocks() as demo:
-    gr.HTML("<center><h2 style='font-size: 22px; text-align: center; color: #007BFF;'>ChatGEO IA</h2><p><b>Tire suas dúvidas, peça sugestões sobre os assuntos da Geografia e muito mais!</b></p></center>")
+    gr.HTML("<center><h2 style='font-size: 22px; text-align: center; color: #007BFF;'>ChatGEO IA</h2><p><b>Tire suas dúvidas, peça sugestões sobre os assuntos da Geografia. Usamos o modelo de IA Mistral-7B-Instruct-v0.3. e muito mais!</b></p></center>")
     iface.render()
     
 demo.queue().launch(show_api=False)