Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from typing import List, Tuple
+from typing import List, Dict
 from gradio_client import Client
 
 def create_chat_app():
@@ -18,6 +18,7 @@ def create_chat_app():
             "max_tokens_label": "Maximum Tokens",
             "temperature_label": "Temperature",
             "top_p_label": "Top-p (Nucleus Sampling)",
+            "message_placeholder": "Type your message here...",
             "info_section": """
             ### ℹ️ Information
             - Model: Llama 3.3 70B Instruct
@@ -25,14 +26,13 @@ def create_chat_app():
             - 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."]
+                "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": {
@@ -48,6 +48,7 @@ def create_chat_app():
             "max_tokens_label": "Máximo de Tokens",
             "temperature_label": "Temperatura",
             "top_p_label": "Top-p (Amostragem Nucleus)",
+            "message_placeholder": "Digite sua mensagem aqui...",
             "info_section": """
             ### ℹ️ Informações
             - Modelo: Llama 3.3 70B Instruct
@@ -55,34 +56,33 @@ def create_chat_app():
             - 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."]
+                "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,
-        chat_history: List[Tuple[str, str]],
+        chat_history: List[Dict],
         system_message: str,
         max_tokens: int,
         temperature: float,
         top_p: float,
         language: str,
-    ) -> Tuple[List[Tuple[str, str]], str]:
+    ):
         try:
             client = Client("aifeifei798/feifei-chat")
             
+            # Format conversation history
             formatted_message = f"{system_message}\n\nConversation history:\n"
-            for user, assistant in chat_history:
-                formatted_message += f"User: {user}\n"
-                formatted_message += f"Assistant: {assistant}\n"
+            for msg in chat_history:
+                formatted_message += f"{msg['role']}: {msg['content']}\n"
             
             formatted_message += f"User: {message}"
             
@@ -99,107 +99,114 @@ def create_chat_app():
                 api_name="/chat"
             )
             
-            chat_history.append((message, response))
-            return chat_history, ""  # Return updated history and empty message
+            # Update chat history in the new format
+            chat_history.extend([
+                {"role": "user", "content": message},
+                {"role": "assistant", "content": response}
+            ])
+            
+            return chat_history, ""
             
         except Exception as e:
             error_msg = TRANSLATIONS[language]["error_message"].format(str(e))
-            chat_history.append((message, error_msg))
+            chat_history.append({"role": "assistant", "content": error_msg})
             return chat_history, ""
 
-    def update_interface(language: str):
-        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["info_section"]
-        )
-
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        # Language selector
-        language = gr.Radio(
-            choices=["en", "pt"],
-            value="en",
-            label="Language/Idioma",
-            interactive=True
-        )
+        current_language = gr.State("en")
         
-        # Dynamic content containers
-        title_md = gr.Markdown(TRANSLATIONS["en"]["title"])
-        description_md = gr.Markdown(TRANSLATIONS["en"]["description"])
+        gr.Markdown(TRANSLATIONS["en"]["title"])
+        gr.Markdown(TRANSLATIONS["en"]["description"])
         
         with gr.Group():
             chatbot = gr.Chatbot(
                 value=[],
                 height=400,
-                label=TRANSLATIONS["en"]["title"]
+                type="messages"  # Use the new messages format
             )
-            msg = gr.Textbox(
-                label="Message",
-                placeholder="Type your message here...",
+            
+            message = gr.Textbox(
+                placeholder=TRANSLATIONS["en"]["message_placeholder"],
                 lines=3
             )
             
-            with gr.Row():
-                system_msg = gr.Textbox(
+            with gr.Accordion("Settings", open=False):
+                system_message = gr.Textbox(
                     value=TRANSLATIONS["en"]["system_message"],
                     label=TRANSLATIONS["en"]["system_message_label"]
                 )
-                max_tokens = gr.Slider(
-                    minimum=1,
-                    maximum=4096,
-                    value=2048,
-                    step=1,
-                    label=TRANSLATIONS["en"]["max_tokens_label"]
-                )
-                temperature = gr.Slider(
-                    minimum=0.1,
-                    maximum=2.0,
-                    value=0.7,
-                    step=0.1,
-                    label=TRANSLATIONS["en"]["temperature_label"]
-                )
-                top_p = gr.Slider(
-                    minimum=0.1,
-                    maximum=1.0,
-                    value=0.95,
-                    step=0.05,
-                    label=TRANSLATIONS["en"]["top_p_label"]
+                
+                with gr.Row():
+                    max_tokens = gr.Slider(
+                        minimum=1,
+                        maximum=4096,
+                        value=2048,
+                        step=1,
+                        label=TRANSLATIONS["en"]["max_tokens_label"]
+                    )
+                    temperature = gr.Slider(
+                        minimum=0.1,
+                        maximum=2.0,
+                        value=0.7,
+                        step=0.1,
+                        label=TRANSLATIONS["en"]["temperature_label"]
+                    )
+                    top_p = gr.Slider(
+                        minimum=0.1,
+                        maximum=1.0,
+                        value=0.95,
+                        step=0.05,
+                        label=TRANSLATIONS["en"]["top_p_label"]
+                    )
+            
+            with gr.Row():
+                language_selector = gr.Radio(
+                    choices=["en", "pt"],
+                    value="en",
+                    label="Language/Idioma",
+                    interactive=True
                 )
-
-        info_md = gr.Markdown(TRANSLATIONS["en"]["info_section"])
+                
+                clear = gr.Button("Clear")
         
-        # Set up chat functionality
-        msg.submit(
-            respond,
-            inputs=[msg, chatbot, system_msg, max_tokens, temperature, top_p, language],
-            outputs=[chatbot, msg]  # Update chat history and clear message input
-        )
+        gr.Markdown(TRANSLATIONS["en"]["info_section"])
         
-        # Examples
         gr.Examples(
             examples=TRANSLATIONS["en"]["examples"],
-            inputs=msg
+            inputs=message
+        )
+        
+        # Event handlers
+        message.submit(
+            respond,
+            [message, chatbot, system_message, max_tokens, temperature, top_p, language_selector],
+            [chatbot, message]
         )
         
-        # Update interface when language changes
-        language.change(
-            fn=update_interface,
-            inputs=[language],
+        clear.click(lambda: ([], ""), outputs=[chatbot, message])
+        
+        # Update interface text when language changes
+        def update_language(lang):
+            trans = TRANSLATIONS[lang]
+            return (
+                trans["message_placeholder"],
+                trans["system_message"],
+                trans["system_message_label"],
+                trans["max_tokens_label"],
+                trans["temperature_label"],
+                trans["top_p_label"]
+            )
+        
+        language_selector.change(
+            update_language,
+            inputs=[language_selector],
             outputs=[
-                title_md,
-                description_md,
-                system_msg,
-                system_msg.label,
-                max_tokens.label,
-                temperature.label,
-                top_p.label,
-                info_md
+                message,
+                system_message,
+                system_message,  # For label
+                max_tokens,      # For label
+                temperature,     # For label
+                top_p           # For label
             ]
         )