initial app

app.py

@@ -3,7 +3,226 @@ import json
 import os
 import uuid
 import streamlit as st
+import math
 from datetime import datetime, timedelta
 from dotenv import load_dotenv
+from typing import List
 
-load_dotenv()
+load_dotenv()
+
+def get_question_topics():
+    url = os.environ["BASE_URL"]+"/question-topics"
+    headers = {
+        "x-api-key": os.environ["API_KEY"]
+    }
+    try:
+        response = requests.get(url,headers=headers)
+        response.raise_for_status()
+        return response.json()["output"]
+    except Exception as e:
+        print(f"Fehler beim Fetchen der Topics: {e}")
+        st.error("Someting went wrong. Please try again later.", icon="🚨") 
+
+def get_coding_task_topics():
+    url = os.environ["BASE_URL"]+"/coding-task-topics"
+    headers = {
+        "x-api-key": os.environ["API_KEY"]
+    }
+    try:
+        response = requests.get(url,headers=headers)
+        response.raise_for_status()
+        return response.json()["output"]
+    except Exception as e:
+        print(f"Fehler beim Fetchen der Topics: {e}")
+        st.error("Someting went wrong. Please try again later.", icon="🚨") 
+
+def get_questions(topic):
+    url = os.environ["BASE_URL"]+"/questions"+f"?dimension={topic}"
+    headers = {
+        "x-api-key": os.environ["API_KEY"]
+    }
+    try:
+        response = requests.get(url, data=topic, headers=headers)
+        response.raise_for_status()
+        # print(response.json())
+        return response.json()["output"]
+    except Exception as e:
+        print(f"Fehler beim Fetchen der Topics: {e}")
+        st.error("Someting went wrong. Please try again later.", icon="🚨")
+
+def get_coding_tasks(topic):
+    url = os.environ["BASE_URL"]+"/coding-tasks"+f"?topic={topic}"
+    headers = {
+        "x-api-key": os.environ["API_KEY"]
+    }
+    try:
+        response = requests.get(url, data=topic, headers=headers)
+        response.raise_for_status()
+        print(response.json())
+        return response.json()["output"]
+    except Exception as e:
+        print(f"Fehler beim Fetchen der Topics: {e}")
+        st.error("Someting went wrong. Please try again later.", icon="🚨")
+
+def load_topics_into_question_list():
+    with st.spinner("Loading questions..."):
+        for topic in st.session_state["loaded_questions"]:
+            if topic["topic"] not in st.session_state["selected_topics"]:
+                st.session_state["loaded_questions"].remove(topic)
+        for topic in st.session_state["selected_topics"]:
+            if topic not in [topic["topic"] for topic in st.session_state["loaded_questions"]]:
+                question_element = {"topic": topic, "questions": get_questions(topic)}
+                st.session_state["loaded_questions"].append(question_element)
+
+def load_topics_into_task_list():
+    with st.spinner("Loading coding tasks..."):
+        for topic in st.session_state["loaded_tasks"]:
+            if topic["topic"] not in st.session_state["selected_task_topics"]:
+                st.session_state["loaded_tasks"].remove(topic)
+        for topic in st.session_state["selected_task_topics"]:
+            if topic not in [topic["topic"] for topic in st.session_state["loaded_tasks"]]:
+                question_element = {"topic": topic, "tasks": get_coding_tasks(topic)}
+                st.session_state["loaded_tasks"].append(question_element)
+
+def add_question_to_interview(question):
+    if question not in st.session_state["questions_for_interview"]:
+        st.session_state["questions_for_interview"].append(question)
+    else:
+        print("Question already in list")
+
+def increment_page():
+    if st.session_state["current_page"] < math.ceil(len(st.session_state["questions_to_display"])/10):
+        st.session_state["current_page"] += 1  
+
+def decrement_page():
+    if st.session_state["current_page"] > 1:
+        st.session_state["current_page"] -= 1
+
+def change_question_type():
+    if st.session_state["question_type"] == "question":
+        st.session_state["question_type"] = "coding_tasks"
+    else:
+        st.session_state["question_type"] = "question"
+
+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
+if "task_topics" not in st.session_state:
+    topics = list(set([topic for topic in get_coding_task_topics() if topic != None]))
+    st.session_state["task_topics"] = topics
+if "loaded_questions" not in st.session_state:
+    st.session_state["loaded_questions"] = []
+if "loaded_tasks" not in st.session_state:
+    st.session_state["loaded_tasks"] = []
+if "selected_topics" not in st.session_state:
+    st.session_state["selected_topics"] = []
+if "questions_to_display" not in st.session_state:
+    st.session_state["questions_to_display"] = []
+if "tasks_to_display" not in st.session_state:
+    st.session_state["tasks_to_display"] = []
+if "questions_for_interview" not in st.session_state:
+    st.session_state["questions_for_interview"] = []
+if "current_page" not in st.session_state:
+    st.session_state["current_page"] = 1
+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"] = []
+
+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")
+
+type_col1, type_col2 = st.columns([1, 1])
+
+with type_col1:
+    st.button("Questions", key="question_button", on_click=change_question_type, disabled=True if st.session_state["question_type"] == "question" else False, use_container_width=True)
+with type_col2:
+    st.button("Coding tasks", key="coding_tasks", on_click=change_question_type, disabled=True if st.session_state["question_type"] == "coding_tasks" else False, use_container_width=True)
+
+
+if st.session_state["question_type"] == "question":
+    st.multiselect("Select the topics for the questions", st.session_state["topics"], key="selected_topics", default=None, on_change=load_topics_into_question_list)
+    st.divider()
+    if len(st.session_state["selected_topics"]) > 0:
+        st.subheader("Questions to choose from:")
+        st.write("Filter according to the difficulty of the questions:")
+        st.checkbox("Easy", key="checkbox_filter_easy", value=True)
+        st.checkbox("Medium", key="checkbox_filter_medium", value=True)
+        st.checkbox("Hard", key="checkbox_filter_hard", value=True)
+        st.write("Filter by the topic of the questions:")
+        for topic in st.session_state["loaded_questions"]:
+            st.checkbox(topic["topic"], key="checkbox_filter_"+topic["topic"], value=True)
+        result_questions = []
+        for i, topic in enumerate(st.session_state["loaded_questions"]):
+            if st.session_state["checkbox_filter_"+topic["topic"]]:
+                for j, question in enumerate(topic["questions"]):
+                    if st.session_state["checkbox_filter_easy"] and question["difficulty"] == "Easy":
+                        result_questions.append(question)
+                    if st.session_state["checkbox_filter_medium"] and question["difficulty"] == "Medium":
+                        result_questions.append(question)
+                    if st.session_state["checkbox_filter_hard"] and question["difficulty"] == "Hard":
+                        result_questions.append(question)
+        st.session_state["questions_to_display"] = result_questions
+        if len(st.session_state["questions_to_display"]) > 0:
+            nav_col1, nav_col2, nav_col3 = st.columns([1, 4, 1])
+
+            nav_col1.button("◀️", key="nav1", use_container_width=True, on_click=decrement_page, disabled=True if st.session_state["current_page"] == 1 else False)
+            nav_col2.markdown(f"<div style='text-align: center;'>Page {str(st.session_state['current_page'])} / {str(math.ceil(len(st.session_state['questions_to_display'])/10))}</div>", unsafe_allow_html=True)
+            nav_col3.button("▶️", key="nav2", use_container_width=True, on_click=increment_page, disabled=True if st.session_state["current_page"] == math.ceil(len(st.session_state["questions_to_display"])/10) else False)
+
+            for i in range((st.session_state["current_page"]-1)*10, st.session_state["current_page"]*10 if st.session_state["current_page"]*10 < len(st.session_state["questions_to_display"]) else len(st.session_state["questions_to_display"])):
+                col_1, col_2 = st.columns([7, 1])
+                with col_1:
+                    st.write(st.session_state["questions_to_display"][i]["question"])
+                with col_2:
+                    st.button("Select", disabled=True if st.session_state["questions_to_display"][i] in st.session_state["questions_for_interview"] else False ,key="select_button_"+str(i), on_click=add_question_to_interview, args=(st.session_state["questions_to_display"][i],))
+else: 
+    st.multiselect("Select the topics for the coding tasks", options=st.session_state["task_topics"], key="selected_task_topics", default=None, on_change=load_topics_into_task_list)      
+    st.divider()
+    if len(st.session_state["selected_task_topics"]) > 0:
+        st.subheader("Coding tasks to choose from:")
+        st.write("Filter by the topic of the coding task:")
+        for topic in st.session_state["loaded_tasks"]:
+            st.checkbox(topic["topic"], key="checkbox_filter_tasks_"+topic["topic"], value=True)
+        result_tasks = []
+        for i, topic in enumerate(st.session_state["loaded_tasks"]):
+            if st.session_state["checkbox_filter_tasks_"+topic["topic"]]:
+                for j, task in enumerate(topic["tasks"]):
+                    result_tasks.append(task)
+        st.session_state["tasks_to_display"] = result_tasks
+        if len(st.session_state["tasks_to_display"]) > 0:
+            nav_col1, nav_col2, nav_col3 = st.columns([1, 4, 1])
+
+            nav_col1.button("◀️", key="nav1", use_container_width=True, on_click=decrement_page, disabled=True if st.session_state["current_page"] == 1 else False)
+            nav_col2.markdown(f"<div style='text-align: center;'>Page {str(st.session_state['current_page'])} / {str(math.ceil(len(st.session_state['tasks_to_display'])/10))}</div>", unsafe_allow_html=True)
+            nav_col3.button("▶️", key="nav2", use_container_width=True, on_click=increment_page, disabled=True if st.session_state["current_page"] == math.ceil(len(st.session_state["tasks_to_display"])/10) else False)
+
+            for i in range((st.session_state["current_page"]-1)*10, st.session_state["current_page"]*10 if st.session_state["current_page"]*10 < len(st.session_state["tasks_to_display"]) else len(st.session_state["tasks_to_display"])):
+                col_1, col_2 = st.columns([7, 1])
+                with col_1:
+                    st.write(st.session_state["tasks_to_display"][i]["task"])
+                with col_2:
+                    st.button("Select", disabled=True if st.session_state["tasks_to_display"][i] in st.session_state["tasks_for_interview"] else False ,key="select_button_tasks_"+str(i), on_click=st.session_state["tasks_for_interview"].append, args=(st.session_state["tasks_to_display"][i],))
+st.divider()
+
+st.subheader("Questions for the interview:")
+if len(st.session_state["questions_for_interview"]) == 0:
+    st.write("No questions selected yet.")
+for i, question in enumerate(st.session_state["questions_for_interview"]):
+    interview_col_1, interview_col_2 = st.columns([7, 1])
+    with interview_col_1:
+        st.write(question["question"])
+    with interview_col_2:
+        st.button("Remove", key="remove_button_"+str(i), on_click=st.session_state["questions_for_interview"].remove, args=(question,))
+st.subheader("Coding tasks for the interview:")
+if len(st.session_state["tasks_for_interview"]) == 0:
+    st.write("No coding tasks selected yet.")
+for i, task in enumerate(st.session_state["tasks_for_interview"]):
+    interview_col_1, interview_col_2 = st.columns([7, 1])
+    with interview_col_1:
+        st.write(task["task"])
+    with interview_col_2:
+        st.button("Remove", key="remove_button_tasks_"+str(i), on_click=st.session_state["tasks_for_interview"].remove, args=(task,))
+st.divider()