Update with adding cvs afterwards

app.py

@@ -167,6 +167,8 @@ if "pdf_data_cvs" not in st.session_state:
     st.session_state["pdf_data_cvs"] = None
 if "pdf_data_cvs_string" not in st.session_state:
     st.session_state["pdf_data_cvs_string"] = None
+if "pdf_data_cvs_names" not in st.session_state:
+    st.session_state["pdf_data_cvs_names"] = []
 if "pdf_data_jobdescription" not in st.session_state:
     st.session_state["pdf_data_jobdescription"] = None
 if "pdf_data_jobdescription_string" not in st.session_state:
@@ -211,6 +213,7 @@ with st.container():
             pdf_data_cvs_string = ""
             for i,cv in enumerate(st.session_state["cvs"]):
                 print(cv.name)
+                st.session_state["pdf_data_cvs_names"].append(cv.name)
                 # print(cv.name)
                 # print(cv.size)
                 cv_data_bytes = cv.read()
@@ -226,8 +229,36 @@ with st.container():
             st.session_state["pdf_data_cvs_string"] = pdf_data_cvs_string
             st.session_state["pdf_data_jobdescription"] = pdf_data_jobdescription
             st.session_state["pdf_data_jobdescription_string"] = pdf_data_jobdescription_string
+        if st.session_state["pdf_data_cvs"]:
+            pdf_data_cvs_string = ""
+            initial_cv_length = len(st.session_state["pdf_data_cvs"])
+            for i,cv in enumerate(st.session_state["cvs"]):
+                if cv.name not in st.session_state["pdf_data_cvs_names"]:
+                    st.session_state["pdf_data_cvs_names"].append(cv.name)
+                    print("At second:"+cv.name)
+                    cv_data_bytes = cv.read()
+                    st.session_state["pdf_data_cvs"].append(cv_data_bytes)
+                    pdf_reader_cvs = PdfReader(io.BytesIO(cv_data_bytes))
+                    pdf_data_cvs_string += "CV "+str(i+1+initial_cv_length)+": "
+                    for page_num in range(len(pdf_reader_cvs.pages)):
+                        page = pdf_reader_cvs.pages[page_num]
+                        pdf_data_cvs_string += page.extract_text()
+                    pdf_data_cvs_string += "\n"
+                    st.session_state["pdf_data_cvs_string"] += pdf_data_cvs_string
+            for i,name in enumerate(st.session_state["pdf_data_cvs_names"]):
+                # print(name)
+                found = False
+                for j,cv in enumerate(st.session_state["cvs"]):
+                    # print(cv.name)
+                    if name == cv.name:
+                        found = True
+                if not found:
+                    print("gelöscht: "+name)
+                    del st.session_state["pdf_data_cvs"][i]
+                    del st.session_state["pdf_data_cvs_names"][i]
         pre_generate = st.toggle("Activate to pre generate questions", key="pre_toggle")
         if pre_generate:
+
             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:
@@ -344,15 +375,12 @@ with st.container():
             print(st.session_state["final_question_string"])
             
             # pdf_data_cv = uploaded_file_cv.read()
-
+            print(len(st.session_state["pdf_data_cvs"]))
+            print(st.session_state["pdf_data_cvs_names"])
             upload_success = upload_blob(pdf_name, json_data, st.session_state["pdf_data_jobdescription"],st.session_state["pdf_data_cvs"],st.session_state["pre_toggle"],st.session_state["final_question_string"])
         else:
             st.write("Please fill out both fields and upload a PDF file.")
 
 
 if not upload_success:
-    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()")
+    st.error('An error has occurred. Please contact the administrator. Sorry for the inconvenience.', icon="🚨")