Update app.py

app.py

@@ -22,11 +22,12 @@ def respond(
         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": "Narrator", "content": val[1]})
+for val in history:
+    if len(val) > 0 and val[0]:  # Verifica se existe um primeiro elemento
+        messages.append({"role": "user", "content": val[0]})
+    if len(val) > 1 and val[1]:  # Verifica se existe um segundo elemento
+        messages.append({"role": "narrator", "content": val[1]})
+
 
     # Adiciona a nova mensagem do usuário
     messages.append({"role": "user", "content": message})
@@ -108,7 +109,7 @@ with gr.Blocks() as demo:
     chat_interface = gr.Interface(
         fn=respond,
         inputs=[chat_input, character, initial_context_message, max_tokens, temperature, top_p],
-        outputs=initial_context_message,
+        outputs=chat_output,
         title="King B",
         description="Story",
     )