Update tools/ai_act_generator.py

tools/ai_act_generator.py

@@ -165,29 +165,40 @@ 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) as demo:
-        gr.HTML("""
-        <style>
-            footer, 
-            .absolute.bottom-4.right-4,
-            a[href*="gradio.app"],
-            a[href*="huggingface.co"],
-            .svelte-drgfjz[data-testid="clear-button"],
-            .svelte-1ipelgc[data-testid="block-settings-button"] {
-                display: none !important;
-            }
-        </style>
-        """)
-
+    with gr.Blocks(
+        fill_height=True,
+        css=".prose > div.flex.justify-between.items-center { display: none !important; }"
+    ) as demo:
         initial_question = QUESTIONS[0][1]
-        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.
+        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)
 
-What is the name of your organization?
-""")
         ...
 
         stored_messages = [initial_message]