Update app.py

app.py

@@ -44,7 +44,7 @@ def generate_questions(model_choice, role, question_type, num_questions, difficu
     elif model_choice == "Gemini":
         llm = get_llm(model_choice)
         response = llm.invoke(prompt)
-        return response.content.strip().split("\n")
+        return response.content.split("\n")
     else:
         raise ValueError("Unsupported model choice.")
 
@@ -210,14 +210,16 @@ st.markdown(
     .stButton>button:hover {
         background-color: #45a049;
     }
-    .progress-container {
-        padding: 15px;
-        background-color: #f1f8e9;
+    .output-container {
+        border: 2px solid #2196F3;
         border-radius: 8px;
-        border: 1px solid #c5e1a5;
-        margin-bottom: 20px;
+        padding: 15px;
+        margin: 15px 0;
+        background-color: #f1f1f1;
     }
-    .output-container {
+    .progress-container {
+        border: 2px solid #2196F3;
+        border-radius: 8px;
         padding: 15px;
         margin: 15px 0;
         background-color: #e3f2fd;
@@ -225,8 +227,6 @@ st.markdown(
     .sidebar {
         background-color: #ffffff; /* Sidebar background color */
         padding: 1em;
-        max-height: 600px; /* Adjust height for scrolling */
-        overflow-y: auto;
     }
     .footer {
         background-color: #4CAF50;
@@ -271,17 +271,7 @@ with st.sidebar:
 st.title("💼 TechPrep")
 
 model_choice = st.selectbox("Select AI Model", ["Gemini", "OpenAI"], key="select_model")
-
-# Updated role dropdown with scrollable feature
-roles = [
-    "Software Developer", "Data Analyst", "Marketing Manager", "Project Manager", "UX Designer",
-    "Machine Learning Engineer", "Data Scientist", "AI Researcher", "Product Manager", "Business Analyst",
-    "Web Developer", "Mobile App Developer", "Cloud Engineer", "DevOps Engineer", "System Analyst",
-    "Database Administrator", "Cybersecurity Specialist", "QA Engineer", "IT Support Specialist", "Network Engineer",
-    "Graphic Designer"
-]
-role = st.selectbox("Select Role", roles, key="select_role")
-
+role = st.selectbox("Select Role", ["GenAI Engineer", "Machine Learning Engineer", "Data Scientist", "Software Developer", "Marketing Manager"], key="select_role")
 question_type = st.selectbox("Select Question Type", ["Behavioral", "Technical", "Situational", "Case Study", "Problem Solving"], key="select_question_type")
 num_questions = st.slider("Number of Questions", 1, 10, key="slider_num_questions")
 difficulty = st.selectbox("Select Difficulty Level", ["Easy", "Medium", "Hard"], key="select_difficulty")
@@ -290,30 +280,23 @@ st.header("📝 Generate Interview Questions")
 if st.button("Generate Questions", key="generate_questions"):
     with st.spinner("Generating questions..."):
         questions = generate_questions(model_choice, role, question_type, num_questions, difficulty)
-        
-        # Paginate questions
-        if "question_index" not in st.session_state:
-            st.session_state.question_index = 0
-        
-        if "questions" not in st.session_state:
-            st.session_state.questions = questions
-        
-        total_questions = len(st.session_state.questions)
-        if total_questions > 0:
-            st.markdown(style_output(f"Question {st.session_state.question_index + 1} of {total_questions}:", "#4CAF50"), unsafe_allow_html=True)
-            st.write(st.session_state.questions[st.session_state.question_index])
+        if questions:
+            question_index = st.session_state.get('question_index', 0)
+            st.session_state['questions'] = questions
+            st.session_state['question_index'] = question_index
+            st.session_state['questions_count'] = len(questions)
             
-            col1, col2 = st.columns(2)
-            with col1:
-                if st.session_state.question_index > 0:
-                    if st.button("Previous Question"):
-                        st.session_state.question_index -= 1
-                        
-            with col2:
-                if st.session_state.question_index < total_questions - 1:
-                    if st.button("Next Question"):
-                        st.session_state.question_index += 1
-                        
+            if question_index < len(questions):
+                st.markdown(style_output("Question:", "#4CAF50"), unsafe_allow_html=True)
+                st.write(st.session_state['questions'][question_index])
+                col1, col2 = st.columns([3, 1])
+                with col2:
+                    if st.button("Next", key="next_question"):
+                        if question_index + 1 < len(st.session_state['questions']):
+                            st.session_state['question_index'] += 1
+                        else:
+                            st.session_state['question_index'] = 0  # Loop back to first question
+                            st.success("You have reached the end of the questions.")
         progress_data["questions_solved"][question_type] += num_questions
 
 st.header("🗣️ Provide Feedback")
@@ -353,3 +336,4 @@ st.markdown("""
     </div>
     """, unsafe_allow_html=True)
 
+