Update app.py

app.py

@@ -133,19 +133,22 @@ with tab3:
     query = st.text_input('Query for A/B Test')
     url = st.text_input('URL for A/B Test')
 
+    if 'step' not in st.session_state:
+        st.session_state.step = 0
+
     if st.button('Scrape A/B'):
         title_A, description_A = extract_title_description(url)
         st.session_state['title_A'] = title_A
         st.session_state['description_A'] = description_A
-        st.write(f'Extracted Title A: {title_A}')
-        st.write(f'Extracted Description A: {description_A}')
-        
+        st.session_state.step = 1
+
+    if st.session_state.step == 1:
         title_B = st.text_input('Title B', value=st.session_state.get('title_A', ''))
         description_B = st.text_area('Description B', value=st.session_state.get('description_A', ''))
 
         if st.button('Predict A/B'):
             if query and url:
-                binary_result_A, confidence_A = predict(query, title_A, description_A, url, fasttext_model)
+                binary_result_A, confidence_A = predict(query, st.session_state['title_A'], st.session_state['description_A'], url, fasttext_model)
                 binary_result_B, confidence_B = predict(query, title_B, description_B, url, fasttext_model)
                 
                 st.write(f'Results for A: Predicted +/-: {binary_result_A}, Conf.: {confidence_A:.2%}')
@@ -157,5 +160,3 @@ with tab3:
                     st.write("B is better than A")
                 else:
                     st.write("B is the same as A")
-            else:
-                st.write('Please enter both a query and a URL.')