added db integration

app.py

@@ -1,18 +1,13 @@
 import requests
-import json
 import os
-import uuid
 import streamlit as st
 import math
 from io import BytesIO
 from reportlab.lib.pagesizes import letter
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, PageTemplate, Frame, Image
-from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
-from reportlab.lib.units import inch
-from reportlab.lib.utils import ImageReader
-from datetime import datetime, timedelta
+from reportlab.platypus import SimpleDocTemplate, Paragraph
+from reportlab.lib.styles import getSampleStyleSheet
+from azure.cosmos import CosmosClient
 from dotenv import load_dotenv
-from typing import List
 
 load_dotenv()
 
@@ -113,6 +108,22 @@ def change_question_type():
     else:
         st.session_state["question_type"] = "question"
 
+def update_questions_and_tasks():
+    with st.spinner("Updating questions and tasks..."):
+        client: CosmosClient = st.session_state["db"]
+        database = client.get_database_client("assessment-database")
+        container = database.get_container_client("assessments")
+        assessment = st.session_state["assessment_db_object"]
+        questions_for_db = []
+        for question in st.session_state["questions_for_interview"]:
+            questions_for_db.append({"question": question["question"], "perfect_answer": question["perfect_answer"], "score": "", "evaluation": ""})
+        assessment["questions"] = questions_for_db
+        tasks_for_db = []
+        for task in st.session_state["tasks_for_interview"]:
+            tasks_for_db.append({"coding_task": task["task"], "perfect_solution": task["perfect_solution"], "score": "", "evaluation": ""})
+        assessment["coding_tasks"] = tasks_for_db
+        container.upsert_item(assessment)
+
 if "topics" not in st.session_state:
     topics = list(set([topic["dimension"] for topic in get_question_topics() if topic["dimension"] != None]))
     st.session_state["topics"] = topics
@@ -137,12 +148,28 @@ if "question_type" not in st.session_state:
     st.session_state["question_type"] = "question"
 if "tasks_for_interview" not in st.session_state:
     st.session_state["tasks_for_interview"] = []
+if "db" not in st.session_state:
+    client = CosmosClient(os.environ["COSMOS_DB_ENDPOINT"], os.environ["COSMOS_DB_KEY"])   
+    st.session_state["db"] = client
+if "assessment_db_object" not in st.session_state:
+    if "assessment-id" in st.query_params:
+        client: CosmosClient = st.session_state["db"]
+        database = client.get_database_client("assessment-database")
+        container = database.get_container_client("assessments")
+        assessment = container.query_items(query="SELECT * FROM c WHERE c.id = @assessment_id", parameters=[dict(name="@assessment_id", value=st.query_params["assessment-id"]) ,], enable_cross_partition_query=True, max_item_count=1)
+        st.session_state["assessment_db_object"] = list(assessment)[0]
+    else:
+        st.error("No assessment found. Please contact the side admin.", icon="🚨")
+        st.stop()
 
 col1, col2 = st.columns([2, 1])
 
 col1.title("Interview Preparation")
 col2.image("https://www.workgenius.com/wp-content/uploads/2023/03/WorkGenius_navy-1.svg")
 
+
+st.write("Assessment: "+st.session_state["assessment_db_object"]["assessment_title"])
+
 type_col1, type_col2 = st.columns([1, 1])
 
 with type_col1:
@@ -261,4 +288,4 @@ if len(st.session_state["questions_for_interview"]) > 0 and len(st.session_state
     pdf_buffer.seek(0)
     
     # Zeige den Download-Button für die PDF an
-    st.download_button(label="Download", data=pdf_buffer, file_name="interview_questions.pdf", mime="application/pdf")
+    st.download_button(label="Download", data=pdf_buffer, file_name=f'interview_questions_{st.session_state["assessment_db_object"]["assessment_title"]}.pdf', mime="application/pdf", on_click=update_questions_and_tasks)

requirements.txt

@@ -1,3 +1,4 @@
 streamlit
 python-dotenv
-reportlab
+reportlab
+azure-cosmos