Updated frontend

app.py

@@ -38,6 +38,13 @@ with colu2:
     st.subheader("Corrected Output")
     gemini_output = st.empty()
 
+# Dropdown to select the type of question
+st.subheader("Select Question Type")
+question_type = st.selectbox(
+    "Choose the type of questions to generate:",
+    ["Logical Questions", "Interview-Based Questions", "Code Analysis Questions"]
+)
+
 # Buttons for different functionalities
 col1, col2, col3 = st.columns([0.3, 0.3, 0.3])
 with col1:
@@ -52,12 +59,16 @@ with col1:
 
 with col2:
     if st.button("Generate Questions"):
-            # Generate questions automatically, displayed below the text boxes
-            generated_questions = generate_questions(code_input)
-            st.subheader("🤖 Model-Generated Questions")
-            st.write(f"Raw Response: {generated_questions}")
+        logger.info(f"Generating {question_type.lower()}.")
+        try:
+            # Generate questions based on user selection
+            generated_questions = generate_questions(code_input, question_type)
+            st.subheader(f"🤖 Model-Generated {question_type}")
+            st.write(generated_questions)
+        except Exception as e:
+            st.error(f"Error: Could not generate questions: {e}")
+            logger.error(f"Question generation error: {e}")
 
-                
 with col3:
     if st.button("Corrected Code"):
         logger.info("User requested code correction.")
@@ -71,10 +82,11 @@ st.session_state.helpful = st.radio("Were the questions helpful?", ("Yes", "No")
 if st.button("Submit Feedback"):
     if st.session_state.helpful is not None:
         try:
-            feedback_collection.insert_one({"helpful": st.session_state.helpful})
+            feedback_collection.insert_one({"helpful": st.session_state.helpful, "question_type": question_type})
             st.success("Feedback submitted successfully.")
         except Exception as e:
             st.error(f"Failed to submit feedback: {e}")
+
 # Hide Streamlit's default menu and style adjustments for a cleaner look
 st.markdown(
     """