refined app structure with appointment toggle

app.py

@@ -5,12 +5,12 @@ import uuid
 import streamlit as st
 from azure.cosmos import CosmosClient
 from datetime import datetime, timedelta
-from ParamClasses import CreateParams, AppointmentDBItem, Attendee
+from ParamClasses import CreateParams, AppointmentDBItem, Attendee, Appointment
 from dotenv import load_dotenv
 
 load_dotenv()
 
-def create_meeting(input_params: CreateParams) -> dict:
+def create_meeting(input_params) -> dict:
     try:
         url = f"{os.environ.get('BASE_API')}/api/create-appointment"
         response = requests.post(url=url,data=input_params)
@@ -19,18 +19,7 @@ def create_meeting(input_params: CreateParams) -> dict:
     except Exception as e: 
         print(f"Fehler beim erstellen des Termins: {str(e)}")
         st.error("Something went wrong, please contact the site admin.", icon="🚨")
-
-def get_summary(input_params) -> dict:
-    print(input_params)
-    try:
-        url = f"{os.environ.get('BASE_API')}/api/create-interview-summary"
-        response = requests.post(url=url,data=input_params)
-        response.raise_for_status()  
-        return response.json()
-    except Exception as e: 
-        print(f"Fehler beim erstellen des Termins: {str(e)}")
-        st.error(f"Something went wrong: {str(e)}", icon="🚨")
-
+        
 def adjust_datetime(original_datetime_str, offset_option):
     # Konvertiere den originalen DateTime-String in ein datetime-Objekt
     original_datetime = datetime.strptime(original_datetime_str, "%Y-%m-%dT%H:%M:%S.%fZ")
@@ -58,27 +47,117 @@ def adjust_datetime(original_datetime_str, offset_option):
     adjusted_datetime_str = adjusted_datetime.strftime("%Y-%m-%dT%H:%M:%S.%fZ")
     return adjusted_datetime_str
 
-def write_appointment_db_object(interview_data: AppointmentDBItem):
-    print("hitesttest")
+def write_interview_db_object(interview_data):
     try:
-        st.session_state["cosmos_db"].create_item(body=interview_data.model_dump())
+        db_client: CosmosClient = st.session_state["cosmos_db"]
+        database = db_client.get_database_client("appointment-database")
+        container = database.get_container_client("appointments")
+        container.create_item(body=interview_data)
     except Exception as e: 
-        print(f"Fehler beim erstellen des DB items: {str(e)}")
+        print(f"Fehler beim erstellen des Appointment DB items: {str(e)}")
         st.error("Something went wrong, please contact the site admin.", icon="🚨")
 
-
+def write_assessment_db_object(interview_data):
+    try:
+        db_client: CosmosClient = st.session_state["cosmos_db"]
+        database = db_client.get_database_client("assessment-database")
+        container = database.get_container_client("assessments")
+        print(interview_data)
+        container.create_item(body=interview_data)
+    except Exception as e: 
+        print(f"Fehler beim erstellen des Assessment DB items: {str(e)}")
+        st.error("Something went wrong, please contact the site admin.", icon="🚨")
+  
+def create_button_handler():
+    with st.spinner("Creating the appointment..."):
+        start_date_utc_str = datetime.strptime(str(st.session_state["date_input"])+" "+str(st.session_state["time_input"]),"%Y-%m-%d %H:%M:%S").strftime("%Y-%m-%dT%H:%M:%S.%fZ")
+        start_date_str = adjust_datetime(start_date_utc_str, st.session_state["time_zone_option"])
+        end_date = datetime.strptime(str(st.session_state["date_input"])+" "+str(st.session_state["time_input"]),"%Y-%m-%d %H:%M:%S") + timedelta(minutes=st.session_state["duration_input"])
+        end_date_utc_str = end_date.strftime("%Y-%m-%dT%H:%M:%S.%fZ")
+        end_date_str = adjust_datetime(end_date_utc_str, st.session_state["time_zone_option"])
+        request_params = json.dumps({
+            "appointment": {
+                "start_time": start_date_str,
+                "end_time": end_date_str
+            },
+            "subject": st.session_state["subject_input"],
+            "recruiter": {
+                "email": "",#st.session_state["recruiter_mail"],
+                "name": ""#st.session_state["recruiter_mail"]
+            },
+            "client": {
+                "email": "",#st.session_state["client_mail"],
+                "name": ""#st.session_state["client_mail"]
+            },
+            "candidate": {
+                "email": "",#st.session_state["candidate_mail"],
+                "name": ""#st.session_state["candidate_mail"]
+            }
+        })
+        appointment_response = create_meeting(request_params)
+        st.session_state["appointment_response"] = appointment_response
+        if st.session_state["assessment_toggle"]:
+            db_item = {
+                "id":str(uuid.uuid4()),
+                "zoom_meeting_id":str(appointment_response["zoom_meeting"]["id"]),
+                "assessment_title":st.session_state["subject_input"],
+                "start_time":start_date_str,
+                "end_time":end_date_str,
+                "meeting_url":appointment_response["zoom_meeting"]["start_url"],
+                "environment":str(["us"]),
+                "process_status":"assessment_scheduled",
+                "recruiter": {
+                    "email": "",#str(st.session_state["recruiter_mail"]),
+                    "name": ""#str(st.session_state["recruiter_mail"])
+                },
+                "client": {
+                    "email": "",#str(st.session_state["client_mail"]),
+                    "name": ""#str(st.session_state["client_mail"])
+                },
+                "candidate": {
+                    "email": "",#str(st.session_state["candidate_mail"]),
+                    "name": ""#str(st.session_state["candidate_mail"])
+                },
+                "interview_transcript": "",
+                "questions": [],
+                "coding_tasks": []
+            }
+            write_assessment_db_object(db_item)
+        else:
+            db_item = {
+                "id":str(uuid.uuid4()),
+                "zoom_meeting_id":str(appointment_response["zoom_meeting"]["id"]),
+                "process_id":str(uuid.uuid4()),
+                "job_title":st.session_state["subject_input"],
+                "start_time":start_date_str,
+                "end_time":end_date_str,
+                "meeting_url":appointment_response["zoom_meeting"]["start_url"],
+                "environment":str(["us"]),
+                "process_status":"interview_scheduled",
+                "recruiter": {
+                    "email": "",#str(st.session_state["recruiter_mail"]),
+                    "name": ""#str(st.session_state["recruiter_mail"])
+                },
+                "client": {
+                    "email": "",#str(st.session_state["client_mail"]),
+                    "name": ""#str(st.session_state["client_mail"])
+                },
+                "candidate": {
+                    "email": "",#str(st.session_state["candidate_mail"]),
+                    "name": ""#str(st.session_state["candidate_mail"])
+                },
+                "summary_recruiter":"",
+                "summary_client":"",
+                "summary_candidate":""
+            }
+            write_interview_db_object(db_item)
+      
 if "appointment_response" not in st.session_state:
     st.session_state["appointment_response"] = None
-if "subject_input" not in st.session_state:
-    st.session_state["subject_input"] = None
 if "cosmos_db" not in st.session_state:
     # Cosmos DB Client erstellen
     client = CosmosClient(os.environ.get('DB_CONNECTION'), os.environ.get('DB_KEY'))
-
-    # Datenbank und Container referenzieren
-    database = client.get_database_client(os.environ.get('DB_NAME'))
-    container = database.get_container_client(os.environ.get('DB_CONTAINER'))
-    st.session_state["cosmos_db"] = container
+    st.session_state["cosmos_db"] = client
     
 col1, col2 = st.columns([2, 1])
 
@@ -87,100 +166,19 @@ col2.image("https://www.workgenius.com/wp-content/uploads/2023/03/WorkGenius_nav
 
 st.write("Please enter the date, time and duration for the appointment to be created.")
 
-selected_date = st.date_input("Date of the appointment",format="DD/MM/YYYY")
+st.date_input("Date of the appointment",format="DD/MM/YYYY", key="date_input")
 now = datetime.now()
 rounded_now = now + timedelta(minutes=30 - now.minute % 30)
-selected_time = st.time_input("Starting time of the appointment", value=rounded_now)
-time_zone = st.selectbox("Please select your time zone (based on UTC time)",options=["+10","+9","+8","+7","+6","+5","+4","+3","+2","+1","0","-1","-2","-3","-4","-5","-6","-7","-8","-9","-10"],index=10, key="time_option")
-selected_duration = st.select_slider("Duration of the appointment in minutes",[15,30,45,60],value=30)
-subject_input = st.text_input("Enter the subject of the meeting")
+st.time_input("Starting time of the appointment", value=rounded_now, key="time_input")
+st.selectbox("Please select your time zone (based on UTC time)",options=["+10","+9","+8","+7","+6","+5","+4","+3","+2","+1","0","-1","-2","-3","-4","-5","-6","-7","-8","-9","-10"],index=10, key="time_zone_option")
+st.select_slider("Duration of the appointment in minutes",[15,30,45,60],value=30, key="duration_input")
+st.text_input("Enter the subject of the meeting", key="subject_input")
 # recruiter_mail = st.text_input("Please enter the mail of the recruiter", key="recruiter_mail")
 # client_mail = st.text_input("Please enter the mail of the client", key="client_mail")
 # candidate_mail = st.text_input("Please enter the mail of the candidate", key="candidate_mail")
+st.toggle("Activate to create an assessment appointment, otherwise a candidate interview is going to be created", key="assessment_toggle")
+st.button("Create appointment", key="create_button", disabled=False if st.session_state["subject_input"] else True, on_click=create_button_handler)
 
-if not st.session_state["appointment_response"]:
-    if st.button("Create appointment") or st.session_state["appointment_response"]:
-        print("nach button appointment")
-        if subject_input:
-            start_date_utc_str = datetime.strptime(str(selected_date)+" "+str(selected_time),"%Y-%m-%d %H:%M:%S").strftime("%Y-%m-%dT%H:%M:%S.%fZ")
-            print(start_date_utc_str)
-            start_date_str = adjust_datetime(start_date_utc_str, st.session_state["time_option"])
-            print(start_date_str)
-            end_date = datetime.strptime(str(selected_date)+" "+str(selected_time),"%Y-%m-%d %H:%M:%S") + timedelta(minutes=selected_duration)
-            end_date_utc_str = end_date.strftime("%Y-%m-%dT%H:%M:%S.%fZ")
-            end_date_str = adjust_datetime(end_date_utc_str, st.session_state["time_option"])
-            with st.spinner("Creating the appointment..."):
-                request_params = json.dumps({
-                    "appointment": {
-                        "start_time": start_date_str,
-                        "end_time": end_date_str
-                    },
-                    "subject": subject_input,
-                    "recruiter": {
-                        "email": "",#st.session_state["recruiter_mail"],
-                        "name": ""#st.session_state["recruiter_mail"]
-                    },
-                    "client": {
-                        "email": "",#st.session_state["client_mail"],
-                        "name": ""#st.session_state["client_mail"]
-                    },
-                    "candidate": {
-                        "email": "",#st.session_state["candidate_mail"],
-                        "name": ""#st.session_state["candidate_mail"]
-                    }
-                })
-                appointment_response = create_meeting(request_params)
-                # appointment_response = False
-                # st.write(appointment_response)
-            if appointment_response:
-                
-                st.success("The appointment was created correctly.")
-                st.write(appointment_response["zoom_meeting"]["start_url"])
-                db_item = AppointmentDBItem(
-                    id=str(uuid.uuid4()),
-                    zoom_meeting_id=str(appointment_response["zoom_meeting"]["id"]),
-                    process_id=str(uuid.uuid4()),
-                    job_title=subject_input,
-                    start_time=start_date_str,
-                    end_time=end_date_str,
-                    meeting_url=appointment_response["zoom_meeting"]["start_url"],
-                    environment=str(["us"]),
-                    process_status="interview_scheduled",
-                    recruiter=Attendee(
-                        email="",#str(st.session_state["recruiter_mail"]),
-                        name=""#str(st.session_state["recruiter_mail"])
-                    ),
-                    client=Attendee(
-                        email="",#str(st.session_state["client_mail"]),
-                        name=""#str(st.session_state["client_mail"])
-                    ),
-                    candidate=Attendee(
-                        email="",#str(st.session_state["candidate_mail"]),
-                        name=""#str(st.session_state["candidate_mail"])
-                    ),
-                    summary_recruiter="",
-                    summary_client="",
-                    summary_candidate=""
-                )
-                write_appointment_db_object(db_item)
-                st.info("Once you have attended the meeting and it has ended, please wait about 10 minutes before requesting the meeting recording as it will take time to become available ", icon="ℹ️")
-                st.session_state["appointment_response"] = appointment_response
-                print(appointment_response["zoom_meeting"]["id"])
-                st.session_state["subject_input"] = subject_input
-                if st.button("Create Interview Summary"):
-                    print("please")
-                    print(st.session_state["appointment_response"]["zoom_meeting"]["id"])
-                    st.rerun()
-                    # print("nach btn summary")
-                    # with st.spinner("Creating the summary..."):
-                    #     summary_response = get_summary({"interview_subject": subject_input})
-                    # if summary_response:
-                    #     st.write(summary_response)
-                             
-        else: 
-            st.warning("Please enter the subject of the meeting")
-else: 
-    with st.spinner("Creating the summary..."):
-        summary_response = get_summary(json.dumps({"meeting_id": str(st.session_state["appointment_response"]["zoom_meeting"]["id"])}))
-        #summary_response = get_summary(json.dumps({"meeting_id": str(85662554678)}))
-    st.write(summary_response)
+if st.session_state["appointment_response"]:
+    st.success("The appointment was created correctly.")
+    st.write(os.environ.get('BASE_API')+"/meeting-consent?redirect="+st.session_state["appointment_response"]["zoom_meeting"]["start_url"].replace("https://us06web.zoom.us/s/", ""))

requirements.txt

@@ -1,4 +1,5 @@
 openai==0.28.1
 streamlit
 python-dotenv
-azure-cosmos
+azure-cosmos
+pydantic