Update tools/ai_act_generator.py

tools/ai_act_generator.py

@@ -165,42 +165,16 @@ Generated by AI Act Assistant.
 ✅ All answers received. Your PDF is ready below:
 👉 Click the download button to save your AI Act compliance register.""", {"done": True, "pdf": pdf_path}, pdf_path
 
+# === UI for Step-by-Step Form ===
 def launch_step_by_step_ui():
-    with gr.Blocks(
-        fill_height=True,
-        css=".prose > div.flex.justify-between.items-center { display: none !important; }"
-    ) as demo:
+    with gr.Blocks(fill_height=True) as demo:
         initial_question = QUESTIONS[0][1]
-        initial_message = gr.ChatMessage(role="assistant", content=f"👋 Welcome!\n\n{initial_question}")
-        stored_messages = [initial_message]
-        
-        chatbot = gr.Chatbot(type="messages", value=stored_messages)
-        msg = gr.Textbox(label="Your answer")
-        state = gr.State()
-        file_output = gr.File(label="Download PDF", visible=True)
-        restart_button = gr.Button("🔁 Restart")
-
-        def chat_logic(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))
-
-            if not file_path or not os.path.isfile(file_path):
-                file_path = None
-            return messages, updated_state, file_path
-
-        def restart_conversation():
-            return [initial_message], {"step": 0, "answers": {}}, None
-
-        msg.submit(chat_logic, [msg, state], [chatbot, state, file_output])
-        restart_button.click(restart_conversation, outputs=[chatbot, state, file_output])
-
-    demo.launch(show_api=False)
-
-        ...
+        initial_message = gr.ChatMessage(role="assistant", content="""
+👋 Welcome! I will guide you through the AI Act compliance form.
+Let's begin with a few questions to generate your compliance register.
 
+What is the name of your organization?
+""")
         stored_messages = [initial_message]
         chatbot = gr.Chatbot(type="messages", value=stored_messages)
         msg = gr.Textbox(label="Your answer")
@@ -231,4 +205,4 @@ def get_questions():
     return QUESTIONS
 
 def run_tool():
-    return launch_step_by_step_ui()
+    return launch_step_by_step_ui()