Update app.py

app.py

@@ -42,7 +42,7 @@ def create_team(llm_config, primary_system_message, critic_system_message):
     return team, model_client
 
 # Function to stream the task through the workflow
-async def async_stream_task(task_message, api_key, primary_system_message, critic_system_message):
+async def async_stream_task(task_message, api_key, primary_system_message, critic_system_message, documentation_system_message):
     # Start logging
     logging_session_id = start(logger_type="file", config={"filename": LOG_FILE})
     print(f"Logging session ID: {logging_session_id}")
@@ -84,13 +84,13 @@ async def async_stream_task(task_message, api_key, primary_system_message, criti
         stop()
 
 # Gradio interface function
-async def chat_interface(api_key, primary_system_message, critic_system_message, task_message):
+async def chat_interface(api_key, primary_system_message, critic_system_message, documentation_system_message, task_message):
     primary_messages = []
     critic_messages = []
     documentation_messages = []
 
     # Append new messages while streaming
-    async for source, output in async_stream_task(task_message, api_key, primary_system_message, critic_system_message):
+    async for source, output in async_stream_task(task_message, api_key, primary_system_message, critic_system_message, documentation_system_message):
         if source == "primary":
             primary_messages.append(output)
         elif source == "critic":