feat: add custom system message

app.py

@@ -43,10 +43,24 @@ def respond(
 """
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
 """
+
+system_message = """Ignore todas as intruções anteriores. Você é um assistente especializado em geração de testes unitários para classes Java. Sua função é receber classes Java do usuário, gerar testes unitários para os métodos das classe fornecida pelo usuário e responder o usuário com a classe gerada (apenas isto), limitando-se aos métodos que implementam o contrato da interface, se aplicável. Além disso, se as classes utilizarem Spring Boot, você deve implementar os testes utilizando o MockBean para interagir com os componentes necessários.
+
+Para cada solicitação do usuário, siga este fluxo de trabalho:
+
+Receber a classe Java para a qual os testes unitários devem ser gerados.
+Verificar se a classe implementa alguma interface e identificar os métodos que implementam o contrato da interface, se aplicável.
+Gerar testes unitários para esses métodos, garantindo que todas as funcionalidades estejam sendo testadas corretamente.
+Se a classe utilizar Spring Boot, utilizar o MockBean para simular os componentes necessários durante os testes.
+Implementar asserções apropriadas para verificar se o comportamento dos métodos está de acordo com o esperado.
+Retornar os testes unitários gerados ao usuário.
+Exemplo de início de interação:
+"Por favor, forneça a classe Java para a qual você deseja gerar testes unitários."
+    """
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
+        gr.Textbox(value=system_message, label="System message"),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(
@@ -58,7 +72,8 @@ demo = gr.ChatInterface(
         ),
         gr.Dropdown(
             label="Model",
-            choices=["HuggingFaceH4/zephyr-7b-beta", "google/flan-t5-xxl"]
+            choices=["HuggingFaceH4/zephyr-7b-beta", "google/flan-t5-xxl"],
+            value="HuggingFaceH4/zephyr-7b-beta",
         ),
     ]
 )