Update to Spinner

app.py

@@ -231,43 +231,58 @@ with st.container():
             system = sys_prompt.format(job=st.session_state["pdf_data_jobdescription_string"], resume=st.session_state["pdf_data_cvs_string"], n=15)
             if not st.session_state["ai_questions"]:
                 try:
-                    st.write("The questions are generated. This may take a short moment...")
-                    res = openai.ChatCompletion.create(
-                        engine="gpt-4",
-                        temperature=0.2,
-                        messages=[
-                            {
-                                "role": "system",
-                                "content": system,
-                            },
-                        ],
-                        )
-                    st.session_state["ai_questions"] = [item for item in res.choices[0]["message"]["content"].split("\n") if len(item) > 0]
-                    for i,q in enumerate(res.choices[0]["message"]["content"].split("\n")):
-                        st.session_state["disable_row_"+str(i)] = False
-                    st.rerun()
+                    # st.write("The questions are generated. This may take a short moment...")
+                    st.info("The questions are generated. This may take a short moment.", icon="ℹ️")
+                    with st.spinner("Loading..."):
+                        res = openai.ChatCompletion.create(
+                            engine="gpt-4",
+                            temperature=0.2,
+                            messages=[
+                                {
+                                    "role": "system",
+                                    "content": system,
+                                },
+                            ],
+                            )
+                        st.session_state["ai_questions"] = [item for item in res.choices[0]["message"]["content"].split("\n") if len(item) > 0]
+                        for i,q in enumerate(res.choices[0]["message"]["content"].split("\n")):
+                            st.session_state["disable_row_"+str(i)] = False
+                        st.rerun()
                 except Exception as e:
                     print(f"Fehler beim generieren der Fragen: {str(e)}")
                     st.error("An error has occurred. Please reload the page or contact the admin.", icon="🚨")
             else:
-                for i,question in enumerate(st.session_state["ai_questions"]):
-                    cols = st.columns([5,1])
-                    with cols[1]:
-                        if st.button("Accept",use_container_width=True,key="btn_accept_row_"+str(i)):
-                            print("accept")
-                            pattern = re.compile(r"^[1-9][0-9]?\.")
-                            questions_length = len(st.session_state["final_question_string"])
-                            question_from_text_area = st.session_state["text_area_"+str(i)]
-                            question_to_append = str(questions_length+1)+"."+re.sub(pattern, "", question_from_text_area)
-                            st.session_state["final_question_string"].append(question_to_append)
-                            st.session_state["disable_row_"+str(i)] = True
-                            st.rerun()
-                        if st.button("Delete",use_container_width=True,key="btn_del_row_"+str(i)):
-                            print("delete")
-                            st.session_state["ai_questions"].remove(question)
-                            st.rerun()
-                    with cols[0]:
-                        st.text_area(label="Question "+str(i+1)+":",value=question,label_visibility="collapsed",key="text_area_"+str(i),disabled=st.session_state["disable_row_"+str(i)])
+                if len(st.session_state["final_question_string"]) <= 0:
+                    for i,question in enumerate(st.session_state["ai_questions"]):
+                        cols = st.columns([5,1])
+                        with cols[1]:
+                            # if st.button("Accept",use_container_width=True,key="btn_accept_row_"+str(i)):
+                            #     print("accept")
+                            #     pattern = re.compile(r"^[1-9][0-9]?\.")
+                            #     questions_length = len(st.session_state["final_question_string"])
+                            #     question_from_text_area = st.session_state["text_area_"+str(i)]
+                            #     question_to_append = str(questions_length+1)+"."+re.sub(pattern, "", question_from_text_area)
+                            #     st.session_state["final_question_string"].append(question_to_append)
+                            #     st.session_state["disable_row_"+str(i)] = True
+                            #     st.rerun() 
+                            if st.button("Delete",use_container_width=True,key="btn_del_row_"+str(i)):
+                                print("delete")
+                                st.session_state["ai_questions"].remove(question)
+                                st.rerun()
+                        with cols[0]:
+                            st.text_area(label="Question "+str(i+1)+":",value=question,label_visibility="collapsed",key="text_area_"+str(i),disabled=st.session_state["disable_row_"+str(i)])
+                    st.write("If you are satisfied with the questions, then accept them. You can still sort them afterwards.")
+                    if st.button("Accept all questions",use_container_width=True,key="accept_all_questions"):
+                        print("accept all")
+                        for i,question in enumerate(st.session_state["ai_questions"]):
+                                print("accept")
+                                pattern = re.compile(r"^[1-9][0-9]?\.")
+                                questions_length = len(st.session_state["final_question_string"])
+                                question_from_text_area = st.session_state["text_area_"+str(i)]
+                                question_to_append = str(questions_length+1)+"."+re.sub(pattern, "", question_from_text_area)
+                                st.session_state["final_question_string"].append(question_to_append)
+                                st.session_state["disable_row_"+str(i)] = True
+                        st.rerun()
                 for i,final_q in enumerate(st.session_state["final_question_string"]):
                     cols_final = st.columns([5,1])
                     with cols_final[1]:
@@ -336,4 +351,8 @@ with st.container():
 
 
 if not upload_success:
-    st.error('An error has occurred. Please contact the administrator. Sorry for the inconvenience.', icon="🚨")
+    st.error('An error has occurred. Please contact the administrator. Sorry for the inconvenience.', icon="🚨")
+# else:
+#     col_submit_btn, col_empty, col_clear_btn = st.columns([1,4, 1])
+#     if col_clear_btn.button("Clear" ,use_container_width=True):
+#         streamlit_js_eval(js_expressions="parent.window.location.reload()")