Update app.py

app.py

@@ -31,8 +31,6 @@ def init_sqlite():
     conn.commit()
     return conn
 
-conn = init_sqlite()
-
 # Create two expanded columns for wider side-by-side text areas
 colu1, colu2 = st.columns([1, 1])  # Both columns have equal width
 
@@ -55,36 +53,46 @@ question_type = st.selectbox(
 
 # Buttons for different functionalities
 col1, col2, col3 = st.columns([0.3, 0.3, 0.3])
-with col1:
-    if st.button("Run Code"):
-        try:
-            output = interpreter.run_code(code_input)
-            st.subheader("✨ Code Output ✨")
-            st.text_area("Execution Output", output, height=600, max_chars=None)
-        except Exception as e:
-            st.error(f"Error: {e}")
-            logger.error(f"Code execution error: {e}")
 
-with col2:
-    if st.button("Generate 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}")
+# Using context manager for database connection
+with init_sqlite() as conn:
+    with col1:
+        if st.button("Run Code"):
+            try:
+                output = interpreter.run_code(code_input)
+                st.subheader("✨ Code Output ✨")
+                st.text_area("Execution Output", output, height=600, max_chars=None)
+            except Exception as e:
+                st.error(f"Error executing code: {e}")
+                logger.error(f"Code execution error: {e}")
+
+    with col2:
+        if st.button("Generate 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.")
-        corrected_code = get_correction_and_comments(code_input)
-        gemini_output.code(corrected_code, language="python")
+    with col3:
+        if st.button("Corrected Code"):
+            logger.info("User requested code correction.")
+            try:
+                corrected_code = get_correction_and_comments(code_input)
+                gemini_output.code(corrected_code, language="python")
+            except Exception as e:
+                st.error(f"Error: Could not generate corrected code: {e}")
+                logger.error(f"Code correction error: {e}")
 
 # Feedback form (outside of the columns, after all content)
 st.subheader("Feedback")
+if st.session_state.helpful is None:
+    st.session_state.helpful = "No"  # default to No if not selected
+
 st.session_state.helpful = st.radio("Were the questions helpful?", ("Yes", "No"))
 
 if st.button("Submit Feedback"):
@@ -113,4 +121,4 @@ st.markdown(
     </style>
     """,
     unsafe_allow_html=True
-)
+)