Update app.py

app.py

@@ -41,71 +41,55 @@ class MistralChatbot:
     def launch_chat(self):
         additional_inputs=[
             gr.Slider(
-                label="Temperature",
+                label="Temperatura",
                 value=0.9,
                 minimum=0.0,
                 maximum=1.0,
                 step=0.05,
                 interactive=True,
-                info="Higher values produce more diverse outputs",
+                info="Valores mais altos produzem saídas mais diversas.",
             ),
             gr.Slider(
-                label="Max new tokens",
+                label="Máximo de Novos Tokens",
                 value=256,
                 minimum=0,
                 maximum=1048,
                 step=64,
                 interactive=True,
-                info="The maximum numbers of new tokens",
+                info="O número máximo de novos tokens.",
             ),
             gr.Slider(
-                label="Top-p (nucleus sampling)",
+                label="Top-p (Amostragem de Núcleo)",
                 value=0.90,
                 minimum=0.0,
                 maximum=1,
                 step=0.05,
                 interactive=True,
-                info="Higher values sample more low-probability tokens",
+                info="Valores mais altos amostram mais tokens de baixa probabilidade.",
             ),
             gr.Slider(
-                label="Repetition penalty",
+                label="Penalidade de Repetição",
                 value=1.2,
                 minimum=1.0,
                 maximum=2.0,
                 step=0.05,
                 interactive=True,
-                info="Penalize repeated tokens",
+                info="Penalize tokens repetidos.",
             )
         ]
 
-        html_interface = gr.Interface(
+        gr.ChatInterface(
             fn=self.generate,
             inputs=["text", "text"] + ["slider"] * 4,
             outputs="text",
-            examples=[
-                ["", ""],
-                ["", ""],
-                [0.9, 256, 0.90, 1.2]
-            ],
-            title="Chatbot Médico Mistral 7B",
-            theme="compact",
-            inputs_width="100%",
-            additional_inputs=additional_inputs
-        )
-
-        custom_html = """
-        <div style="padding: 10px; background-color: #f8f9fa;">
-            <p style="font-size: 18px; font-weight: bold;">Chat com Mistral 7B</p>
-            <p>Insira sua mensagem e ajuste as configurações:</p>
-        </div>
-        """
-
-        html_interface.configure(
-            custom_html=custom_html,
-            url="/mistral_chat"
-        )
-
-        html_interface.launch(inline=False, inbrowser=True, debug=True)
+            title="Mistral 7B - Chatbot Médico",
+            inputs_width="50%",
+            additional_inputs=additional_inputs,
+            theme="light",  # Alteração para um tema mais claro
+            show_typing_indicator=True,
+            show_timestamps=True,
+            show_multioutput=True
+        ).launch(show_api=False)
 
 # Exemplo de uso:
 if __name__ == "__main__":