Update app.py

app.py

@@ -184,157 +184,65 @@ def connect_resources():
             else:
                 st.success("Thank you for contacting us! We will get back to you soon.")
 
-def style_output(text, color):
-    return f'<div class="output-container"><span style="color: {color}; font-weight: bold;">{text}</span></div>'
-
-# Streamlit app layout
-st.set_page_config(page_title="TechPrep", layout="wide")
-st.markdown(
-    """
-    <style>
-    body {
-        background-color: #e0f7fa; /* Light cyan background color */
-        font-family: Arial, sans-serif;
-    }
-    .stButton>button {
-        width: 100%;
-        height: 3em;
-        font-size: 1.2em;
-        color: white;
-        background-color: #4CAF50;
-        border: none;
-        border-radius: 8px;
-        cursor: pointer;
-        transition: background-color 0.3s ease;
-    }
-    .stButton>button:hover {
-        background-color: #45a049;
-    }
-    .output-container {
-        border: 2px solid #2196F3;
-        border-radius: 8px;
-        padding: 15px;
-        margin: 15px 0;
-        background-color: #f1f1f1;
-    }
-    .progress-container {
-        border: 2px solid #2196F3;
-        border-radius: 8px;
-        padding: 15px;
-        margin: 15px 0;
-        background-color: #e3f2fd;
-    }
-    .sidebar {
-        background-color: #ffffff; /* Sidebar background color */
-        padding: 1em;
-    }
-    .footer {
-        background-color: #4CAF50;
-        color: white;
-        text-align: center;
-        padding: 1em;
-        position: fixed;
-        bottom: 0;
-        width: 100%;
-        border-top: 2px solid #ffffff;
-    }
-    </style>
-    """,
-    unsafe_allow_html=True
-)
-
-# Show welcome message with an icon for 3-4 seconds
-welcome_message = st.empty()
-with welcome_message.container():
-    st.markdown("""
-        <div style="
-            text-align: center;
-            padding: 20px;
-            background-color: #4CAF50;
-            color: white;
-            border-radius: 8px;
-            font-size: 24px;
-        ">
-            <i class="fa fa-smile-o" aria-hidden="true" style="font-size: 40px;"></i> Welcome to TechPrep!
-        </div>
-    """, unsafe_allow_html=True)
-time.sleep(4)  # Wait for 4 seconds
-welcome_message.empty()  # Remove the welcome message
-
-# Sidebar for additional links
-with st.sidebar:
-    st.title("Menu")
-    st.markdown("<h3>About Us</h3><p>We help you prepare for job interviews with simulated questions and feedback.</p>", unsafe_allow_html=True)
-    st.markdown("<h3>Sponsored By</h3><p>Your trusted career preparation partner.</p>", unsafe_allow_html=True)
-    st.markdown("<h3>Contact Us</h3><p>For support, email us at <a href='mailto:support@example.com'>support@example.com</a>.</p>", unsafe_allow_html=True)
-
-st.title("💼 TechPrep")
-
-model_choice = st.selectbox("Select AI Model", ["Gemini", "OpenAI"], key="select_model")
-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")
-
-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)
-        if questions:
-            st.session_state['questions'] = questions
-            st.session_state['question_index'] = 0  # Reset index to start
-            st.session_state['questions_count'] = len(questions)
-            
-            st.markdown(style_output("Question:", "#4CAF50"), unsafe_allow_html=True)
-            st.write(st.session_state['questions'][st.session_state['question_index']])
-            
-            col1, col2 = st.columns([3, 1])
+def display_questions(questions):
+    if questions:
+        for i, question in enumerate(questions):
+            st.write(f"**Question {i+1}:** {question}")
+
+def main():
+    st.set_page_config(page_title="TechPrep", layout="wide")
+    st.title("TechPrep: Interview Preparation Tool")
+
+    # Initialize session state for questions
+    if 'questions' not in st.session_state:
+        st.session_state.questions = []
+    if 'question_index' not in st.session_state:
+        st.session_state.question_index = 0
+
+    # Define sidebar options
+    options = ["Generate Questions", "Track Progress", "Connect with Resources", "Schedule Mock Interview"]
+    selection = st.sidebar.radio("Choose an Option", options)
+
+    if selection == "Generate Questions":
+        model_choice = st.selectbox("Select Model", ["OpenAI", "Gemini"])
+        role = st.text_input("Role")
+        question_type = st.selectbox("Question Type", ["Behavioral", "Technical", "Situational", "Case Study", "Problem Solving"])
+        num_questions = st.number_input("Number of Questions", min_value=1, max_value=10, value=1)
+        difficulty = st.selectbox("Difficulty", ["Easy", "Medium", "Hard"])
+
+        if st.button("Generate Questions"):
+            if role:
+                questions = generate_questions(model_choice, role, question_type, num_questions, difficulty)
+                st.session_state.questions = questions
+                st.session_state.question_index = 0
+                st.success(f"Questions generated successfully!")
+            else:
+                st.error("Please enter a role.")
+
+        # Navigation for questions
+        if st.session_state.questions:
+            display_questions([st.session_state.questions[st.session_state.question_index]])
+
+            col1, col2 = st.columns([1, 1])
+            with col1:
+                if st.session_state.question_index > 0:
+                    if st.button("Previous"):
+                        st.session_state.question_index -= 1
+                        st.experimental_rerun()
             with col2:
-                if st.button("Next", key="next_question"):
-                    if st.session_state['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.")
-                    st.markdown(style_output("Question:", "#4CAF50"), unsafe_allow_html=True)
-                    st.write(st.session_state['questions'][st.session_state['question_index']])
-            
-            progress_data["questions_solved"][question_type] += num_questions
+                if st.session_state.question_index < len(st.session_state.questions) - 1:
+                    if st.button("Next"):
+                        st.session_state.question_index += 1
+                        st.experimental_rerun()
 
-st.header("🗣️ Provide Feedback")
-answer = st.text_area("Submit Your Answer", key="text_area_answer")
-if st.button("Submit Answer", key="submit_answer"):
-    if not answer:
-        st.error("Please enter an answer to receive feedback.")
-    else:
-        with st.spinner("Providing feedback..."):
-            feedback = provide_feedback(model_choice, answer)
-            st.markdown(style_output("Feedback Received:", "#FF5722"), unsafe_allow_html=True)
-            st.write(feedback)
-            progress_data["feedback_provided"] += 1
+    elif selection == "Track Progress":
+        track_progress()
 
-st.header("💡 Interview Tips")
-if st.button("Get Tips", key="get_tips"):
-    with st.spinner("Fetching tips..."):
-        tips = get_tips(model_choice, role)
-        st.markdown(style_output("Tips Received:", "#2196F3"), unsafe_allow_html=True)
-        st.write(tips)
-        progress_data["tips_retrieved"] += 1
+    elif selection == "Connect with Resources":
+        connect_resources()
 
-st.header("📅 Schedule Mock Interview")
-schedule_mock_interview()
-
-st.header("📈 Progress and Resources")
-col1, col2 = st.columns(2)
-with col1:
-    track_progress()
-with col2:
-    connect_resources()
-
-# Footer
-st.markdown("""
-    <div class="footer">
-        <p>&copy; 2024 TechPrep. All rights reserved.</p>
-    </div>
-    """, unsafe_allow_html=True)
+    elif selection == "Schedule Mock Interview":
+        schedule_mock_interview()
 
+if __name__ == "__main__":
+    main()