Update with Sorting Final Questions

app.py

@@ -12,6 +12,11 @@ import time
 import calendar
 import re
 
+openai.api_key = os.getenv("OPENAI_API_KEY")
+openai.api_base = "https://tensora-oai.openai.azure.com/"
+openai.api_type = "azure"
+openai.api_version = "2023-05-15"
+
 connection_string = os.getenv("CONNECTION")
 blob_service_client = BlobServiceClient.from_connection_string(connection_string)
 
@@ -31,7 +36,11 @@ def upload_blob(pdf_name, json_data, pdf_data_jobdescription,pdf_data_cvs, pre_g
         pdf_blob_client.upload_blob(pdf_data_jobdescription, overwrite=True)
 
         upload_job_db_item(pdf_name,len(pdf_data_cvs),json.loads(json_data),pre_generated_bool, custom_questions)
-
+        if pre_generated_bool:
+            for i,question in enumerate(custom_questions):
+                question_nr_for_id = i+1
+                question_id = pdf_name + "-question-nr-" + str(question_nr_for_id)+str(calendar.timegm(time.gmtime()))
+                upload_question_db_item(question_id, pdf_name, question,st.session_state["pdf_data_jobdescription_string"])
         links = []
         names = []
         for i,cv in enumerate(pdf_data_cvs):
@@ -115,6 +124,28 @@ def upload_db_item(name, data, job_description_id, cv_id):
     except Exception as e:
         print(f"Allgemeiner Fehler: {str(e)}")
 
+def upload_question_db_item(id, job_id, question, job_content):
+    endpoint = "https://wg-candidate-data.documents.azure.com:443/"
+    key = os.getenv("CONNECTION_DB")
+    client = CosmosClient(endpoint, key)
+    database = client.get_database_client("ToDoList")
+    container = database.get_container_client("Questions")
+    question_item = {
+        "id": id,
+        "partitionKey" : "wg-question-data-v1",
+        "job_id": job_id,
+        "question_content": question,
+        "job_description": job_content,
+    }
+    try:
+        # Fügen Sie das Element in den Container ein
+        container.create_item(body=question_item)
+        print("Eintrag erfolgreich in die Cosmos DB eingefügt. Container: Questions(Question Data)")
+    except exceptions.CosmosHttpResponseError as e:
+        print(f"Fehler beim Schreiben in die Cosmos DB: {str(e)}")
+    except Exception as e:
+        print(f"Allgemeiner Fehler: {str(e)}")
+
 st.markdown(
 """
 <style>
@@ -143,6 +174,9 @@ if "pdf_data_jobdescription_string" not in st.session_state:
 if "final_question_string" not in st.session_state:
     st.session_state["final_question_string"] = []
 
+def adjust_numbering(lst):
+    return [f"{i + 1}. {item.split('. ', 1)[1]}" for i, item in enumerate(lst)]
+
 with open("sys_prompt_frontend.txt") as f:
     sys_prompt = f.read()
 
@@ -199,7 +233,7 @@ with st.container():
                 try:
                     st.write("The questions are generated. This may take a short moment...")
                     res = openai.ChatCompletion.create(
-                        model="gpt-4",
+                        engine="gpt-4",
                         temperature=0.2,
                         messages=[
                             {
@@ -208,7 +242,7 @@ with st.container():
                             },
                         ],
                         )
-                    st.session_state["ai_questions"] = res.choices[0]["message"]["content"].split("\n")
+                    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()
@@ -234,6 +268,25 @@ with st.container():
                             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)])
+                for i,final_q in enumerate(st.session_state["final_question_string"]):
+                    cols_final = st.columns([5,1])
+                    with cols_final[1]:
+                        if st.button("Up",use_container_width=True,key="btn_up_row_"+str(i),disabled=True if i == 0 else False):
+                            if i > 0:
+                                # Tausche das aktuelle Element mit dem vorherigen Element
+                                st.session_state.final_question_string[i], st.session_state.final_question_string[i - 1] = \
+                                    st.session_state.final_question_string[i - 1], st.session_state.final_question_string[i]
+                                st.session_state.final_question_string = adjust_numbering(st.session_state.final_question_string)
+                                st.rerun()
+                        if st.button("Down",use_container_width=True,key="btn_down_row_"+str(i), disabled=True if i == len(st.session_state["final_question_string"])-1 else False):
+                            if i < len(st.session_state.final_question_string) - 1:
+                                # Tausche das aktuelle Element mit dem nächsten Element
+                                st.session_state.final_question_string[i], st.session_state.final_question_string[i + 1] = \
+                                    st.session_state.final_question_string[i + 1], st.session_state.final_question_string[i]
+                                st.session_state.final_question_string = adjust_numbering(st.session_state.final_question_string)
+                                st.rerun()
+                    with cols_final[0]:
+                        st.write(final_q)
         else:
             with st.expander("Enter up to three predefined questions if needed. Otherwise leave it blank:"):
                     question_one = st.text_input("Enter the first question:")
@@ -283,8 +336,4 @@ with st.container():
 
 
 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="🚨")