Update Gradio_UI.py

Gradio_UI.py

@@ -68,7 +68,7 @@ def step_by_step_agent(user_input, state):
     if step < len(QUESTIONS):
         next_question = QUESTIONS[step][1]
         state["step"] += 1
-        return next_question, state
+        return next_question, state, None
 
     filled_template = f"""
 # AI Act Compliance Register
@@ -124,18 +124,13 @@ def launch_step_by_step_ui():
         restart_button = gr.Button("🔁 Restart")
 
         def chat_logic(user_msg, state):
-            reply, updated_state = step_by_step_agent(user_msg, state)
+            reply, updated_state, file_path = step_by_step_agent(user_msg, state)
             messages = [gr.ChatMessage(role="user", content=user_msg)]
 
             if isinstance(reply, str):
                 messages.append(gr.ChatMessage(role="assistant", content=reply))
 
-            file_path = None
-            if isinstance(updated_state, dict) and updated_state.get("done"):
-                file_path = updated_state["pdf"]
-                return messages, updated_state, file_path
-
-            return messages, updated_state, None
+            return messages, updated_state, file_path
 
         def restart_conversation():
             return [initial_message], {"step": 0, "answers": {}}, None