Update app.py

app.py

@@ -19,14 +19,14 @@ def respond(
     
     # Adiciona a mensagem de contexto automaticamente no início do chat
     if not history:  # Se o histórico estiver vazio, adiciona a mensagem inicial
-        messages.append({"role": "assistant", "content": initial_context_message})
+        messages.append({"role": "Narrator", "content": initial_context_message})
     
     # Processa o histórico para adicionar ao contexto
     for val in history:
         if val[0]:
             messages.append({"role": "user", "content": val[0]})
         if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
+            messages.append({"role": "Narrator", "content": val[1]})
 
     # Adiciona a nova mensagem do usuário
     messages.append({"role": "user", "content": message})
@@ -35,15 +35,12 @@ def respond(
 
 
     # Chama a API do modelo de linguagem para gerar a resposta
-    try:
-        response = client.chat_completion(
-            messages=messages,
-            max_tokens=max_tokens,
-            temperature=temperature,
-            top_p=top_p,
-        ).choices[0].message.content
-    except Exception as e:
-        response = f"Error: {e}"
+    response = client.chat_completion(
+        messages=messages,
+        max_tokens=max_tokens,
+        temperature=temperature,
+        top_p=top_p,
+    ).choices[0].message.content
 
     return response
 
@@ -53,49 +50,22 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 """
 
 
-with gr.Blocks() as demo:
-    # Campo para configuração do contexto inicial
-    initial_context_message = gr.Textbox(
-        value="Long ago, the world of Fabe was a harmonious realm where magic and mortal life existed in perfect balance. This world was protected by four sacred crystals, each tied to one of the elements—fire, water, earth, and air. These crystals were safeguarded by ancient creatures known as the Elders, wise beings who had lived since the dawn of Fabe. However, peace was shattered when an ambitious sorcerer named Zareth discovered a forbidden spell that could harness the power of all four crystals. Driven by his thirst for power, he unleashed this spell, hoping to become a god among mortals. Instead, his actions fractured the world, creating a series of isolated realms, each bound to one elemental crystal but severed from the others. The Elders disappeared, and with them went the knowledge of the true, unified Arthenor. Centuries have passed since the Sundering, and most inhabitants of these realms have no memory of a world beyond their borders. Legends tell of a hero who would rise to restore Arthenor, but no one believes in these old tales anymore. The realms have become self-sustaining but are also plagued by strange creatures, corrupted by elemental imbalances. You are a young adventurer from the Ember Plains, the realm of fire, who stumbles upon a shard of the lost crystal of water. This discovery sets off a chain of events that reveal your latent magical abilities and hint at the fractured truth of Arthenor’s past. You are soon joined by allies from the other realms, each with their own skills and backgrounds, united by a common goal: to restore balance to the realms and stop Zareth, who has been resurrected by his followers.",
-        label="Story"
-    )
-    
-    # Campo para definir o personagem
-    character = gr.Textbox(
-        value="Ben, a young boy",
-        label="Character"
-    )
-    
-    # Sliders para controlar parâmetros do modelo
-    max_tokens = gr.Slider(
-        minimum=2048, 
-        maximum=2048, 
-        value=2048, 
-        step=1, 
-        label="Max new tokens"
-    )
-    temperature = gr.Slider(
-        minimum=0.6, 
-        maximum=0.6, 
-        value=0.6, 
-        step=0.1, 
-        label="Temperature"
-    )
-    top_p = gr.Slider(
-        minimum=0.95, 
-        maximum=0.95, 
-        value=0.95, 
-        step=0.05, 
-        label="Top-p (nucleus sampling)"
-    )
-
-    # Interface de chat com as entradas adicionais visíveis
-    chat_interface = gr.ChatInterface(
-        fn=respond,
-        additional_inputs=[character, max_tokens, temperature, top_p, initial_context_message],
-        title="King B",
-        description="Story",
-    )
+demo = gr.ChatInterface(
+    respond,
+    additional_inputs=[
+        gr.Textbox(value="Este é o contexto do chat. Você é um assistente que ajuda o usuário a explorar ideias e responder perguntas.", label="Initial Context Message"),
+        gr.Textbox(value="Ben, a young boy", label="Character"),
+        gr.Slider(minimum=2048, maximum=2048, value=2048, step=1, label="Max new tokens"),
+        gr.Slider(minimum=0.6, maximum=0.6, value=0.6, step=0.1, label="Temperature"),
+        gr.Slider(
+            minimum=0.95,
+            maximum=0.95,
+            value=0.95,
+            step=0.05,
+            label="Top-p (nucleus sampling)",
+        ),
+    ],
+)
 
 if __name__ == "__main__":
     demo.launch()