Update app.py

app.py

@@ -3,6 +3,74 @@ from typing import List, Tuple
 from gradio_client import Client
 
 def create_chat_app():
+    # Language configurations
+    TRANSLATIONS = {
+        "en": {
+            "title": "🤖 Chat with Llama 3.3 70B",
+            "description": """
+            This is a chatbot based on the Llama 3.3 70B model. To use:
+            1. Type your message in the field below
+            2. Adjust parameters as needed
+            3. Press Enter to send
+            """,
+            "system_message": "You are a helpful and friendly assistant based on the Llama 3.3 70B model.",
+            "system_message_label": "System Message",
+            "max_tokens_label": "Maximum Tokens",
+            "temperature_label": "Temperature",
+            "top_p_label": "Top-p (Nucleus Sampling)",
+            "chat_title": "Chat with Llama 3.3 70B",
+            "chat_description": "An interactive chatbot using the Llama 3.3 70B Instruct model.",
+            "info_section": """
+            ### ℹ️ Information
+            - Model: Llama 3.3 70B Instruct
+            - Language: English/Portuguese
+            - Hosting: Hugging Face Spaces
+            
+            For best performance, adjust the parameters according to your needs.
+            This model has advanced natural language processing capabilities.
+            """,
+            "error_message": "Sorry, an error occurred: {}\nPlease check your connection and settings.",
+            "examples": [
+                ["Hello! How are you?"],
+                ["Can you explain what artificial intelligence is?"],
+                ["What is the capital of Brazil?"],
+                ["Help me write a Python code to calculate Fibonacci."]
+            ]
+        },
+        "pt": {
+            "title": "🤖 Chat com Llama 3.3 70B em Português",
+            "description": """
+            Este é um chatbot baseado no modelo Llama 3.3 70B. Para usar:
+            1. Digite sua mensagem no campo abaixo
+            2. Ajuste os parâmetros conforme necessário
+            3. Pressione Enter para enviar
+            """,
+            "system_message": "Você é um assistente amigável e prestativo que responde em português. Você é baseado no modelo Llama 3.3 70B.",
+            "system_message_label": "Mensagem do Sistema",
+            "max_tokens_label": "Máximo de Tokens",
+            "temperature_label": "Temperatura",
+            "top_p_label": "Top-p (Amostragem Nucleus)",
+            "chat_title": "Chat com Llama 3.3 70B",
+            "chat_description": "Um chatbot interativo usando o modelo Llama 3.3 70B Instruct.",
+            "info_section": """
+            ### ℹ️ Informações
+            - Modelo: Llama 3.3 70B Instruct
+            - Idioma: Português/Inglês
+            - Hospedagem: Hugging Face Spaces
+            
+            Para melhor desempenho, ajuste os parâmetros de acordo com suas necessidades.
+            Este modelo possui capacidades avançadas de processamento de linguagem natural.
+            """,
+            "error_message": "Desculpe, ocorreu um erro: {}\nPor favor, verifique sua conexão e configurações.",
+            "examples": [
+                ["Olá! Como você está?"],
+                ["Pode me explicar o que é inteligência artificial?"],
+                ["Qual é a capital do Brasil?"],
+                ["Me ajude a escrever um código em Python para calcular fibonacci."]
+            ]
+        }
+    }
+
     def respond(
         message: str,
         history: List[Tuple[str, str]],
@@ -10,15 +78,14 @@ def create_chat_app():
         max_tokens: int,
         temperature: float,
         top_p: float,
+        language: str,
     ) -> str:
         """
         Process user message and generate a response using the Llama 3.3 70B model.
         """
         try:
-            # Initialize client for the newer API
             client = Client("aifeifei798/feifei-chat")
             
-            # Format the conversation history and current message
             formatted_message = f"{system_message}\n\nConversation history:\n"
             for user, assistant in history:
                 if user:
@@ -28,13 +95,11 @@ def create_chat_app():
             
             formatted_message += f"User: {message}"
             
-            # Prepare the message payload
             message_payload = {
                 "text": formatted_message,
                 "files": []
             }
             
-            # Get response from the model with the specified parameters
             response = client.predict(
                 message=message_payload,
                 feifei_select=True,
@@ -46,68 +111,106 @@ def create_chat_app():
             return response
             
         except Exception as e:
-            return f"Desculpe, ocorreu um erro: {str(e)}\nPor favor, verifique sua conexão e configurações."
+            return TRANSLATIONS[language]["error_message"].format(str(e))
+
+    def update_interface(language: str):
+        """
+        Update the interface language based on user selection
+        """
+        trans = TRANSLATIONS[language]
+        return (
+            trans["title"],
+            trans["description"],
+            trans["system_message"],
+            trans["system_message_label"],
+            trans["max_tokens_label"],
+            trans["temperature_label"],
+            trans["top_p_label"],
+            trans["chat_title"],
+            trans["chat_description"],
+            trans["info_section"],
+            trans["examples"]
+        )
 
-    # Interface configuration
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        gr.Markdown("""
-        # 🤖 Chat com Llama 3.3 70B em Português
+        # Language selector
+        language = gr.Radio(
+            choices=["en", "pt"],
+            value="en",
+            label="Language/Idioma",
+            interactive=True
+        )
         
-        Este é um chatbot baseado no modelo Llama 3.3 70B. Para usar:
-        1. Digite sua mensagem no campo abaixo
-        2. Ajuste os parâmetros conforme necessário
-        3. Pressione Enter para enviar
-        """)
+        # Dynamic content containers
+        title_md = gr.Markdown()
+        description_md = gr.Markdown()
+        info_md = gr.Markdown()
         
         chatbot = gr.ChatInterface(
             respond,
             additional_inputs=[
                 gr.Textbox(
-                    value="Você é um assistente amigável e prestativo que responde em português. Você é baseado no modelo Llama 3.3 70B.",
-                    label="Mensagem do Sistema"
+                    label="System Message",
+                    value=TRANSLATIONS["en"]["system_message"]
                 ),
                 gr.Slider(
                     minimum=1,
                     maximum=4096,
                     value=2048,
                     step=1,
-                    label="Máximo de Tokens"
+                    label="Maximum Tokens"
                 ),
                 gr.Slider(
                     minimum=0.1,
                     maximum=2.0,
                     value=0.7,
                     step=0.1,
-                    label="Temperatura"
+                    label="Temperature"
                 ),
                 gr.Slider(
                     minimum=0.1,
                     maximum=1.0,
                     value=0.95,
                     step=0.05,
-                    label="Top-p (Amostragem Nucleus)"
+                    label="Top-p"
                 ),
+                language  # Pass the selected language to the respond function
             ],
-            title="Chat com Llama 3.3 70B",
-            description="Um chatbot interativo usando o modelo Llama 3.3 70B Instruct.",
-            examples=[
-                ["Olá! Como você está?"],
-                ["Pode me explicar o que é inteligência artificial?"],
-                ["Qual é a capital do Brasil?"],
-                ["Me ajude a escrever um código em Python para calcular fibonacci."]
-            ]
+            examples=TRANSLATIONS["en"]["examples"]
         )
         
-        gr.Markdown("""
-        ### ℹ️ Informações
-        - Modelo: Llama 3.3 70B Instruct
-        - Idioma: Português
-        - Hospedagem: Hugging Face Spaces
+        # Update interface when language changes
+        language.change(
+            fn=update_interface,
+            inputs=[language],
+            outputs=[
+                title_md,
+                description_md,
+                chatbot.textbox,
+                *[input.label for input in chatbot.additional_inputs[:-1]],  # Exclude language selector
+                chatbot.title,
+                chatbot.description,
+                info_md,
+                chatbot.examples
+            ]
+        )
         
-        Para melhor desempenho, ajuste os parâmetros de acordo com suas necessidades.
-        Este modelo possui capacidades avançadas de processamento de linguagem natural.
-        """)
-    
+        # Initialize interface with default language
+        demo.load(
+            fn=update_interface,
+            inputs=[language],
+            outputs=[
+                title_md,
+                description_md,
+                chatbot.textbox,
+                *[input.label for input in chatbot.additional_inputs[:-1]],
+                chatbot.title,
+                chatbot.description,
+                info_md,
+                chatbot.examples
+            ]
+        )
+
     return demo
 
 if __name__ == "__main__":