Update app.py

app.py

@@ -47,10 +47,11 @@ if html_content:
         st.write("Please paste the HTML piece code, not the URL")
         html_content = None
 
-extract_button = st.button("Extract data format")
+extract_button = st.button("Generate output format & code")
 
 if html_content and extract_button:
     try:
+        st.write("1/2: Generating the output format...")
         output = gpt_assistant.chain_response_format(html_content)
         st.session_state['output_format'] = output
     except NameError:
@@ -61,16 +62,16 @@ if html_content and extract_button:
 if 'output_format' in st.session_state:
     output_format = st.code(st.session_state['output_format'], language="json")
     
-    if st.button("Generate the code"):
-        try:
-            python_code = gpt_assistant.chain_code_generator(st.session_state['output_format'], html_content)
-            st.session_state['code_generated'] = python_code
-            st.session_state['code_generated_exec'] = python_code + "\nresult = extract_info(html_data)"
-
-        except NameError:
-            st.write("Complete the API key field")
-        except AuthenticationError:
-            st.write("Invalid API key")
+    try:
+        st.write("2/2: Generating the code...")
+        python_code = gpt_assistant.chain_code_generator(st.session_state['output_format'], html_content)
+        st.session_state['code_generated'] = python_code
+        st.session_state['code_generated_exec'] = python_code + "\nresult = extract_info(html_data)"
+
+    except NameError:
+        st.write("Complete the API key field")
+    except AuthenticationError:
+        st.write("Invalid API key")
             
 @traceable(run_type="tool")
 def test_the_code(code, full_content):