Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
+import sqlite3
 from Backend import get_correction_and_comments, generate_questions
 from logger import logger
-from pymongo import MongoClient, errors
 import interpreter
 from dotenv import load_dotenv
 import os
@@ -16,14 +16,22 @@ st.title("✨ Interactive Code Assistant with Python Interpreter ✨")
 if "helpful" not in st.session_state:
     st.session_state.helpful = None  # Initialize feedback as None
 
-# Set up MongoDB connection
-try:
-    client = MongoClient(os.getenv("MONGO_URI"), serverSelectionTimeoutMS=5000)
-    client.server_info()  # Trigger an exception if MongoDB is not reachable
-    db = client["Capstone"]
-    feedback_collection = db["Feedback"]
-except errors.ServerSelectionTimeoutError:
-    st.error("Error: Could not connect to MongoDB. Please ensure MongoDB is running.")
+# Set up SQLite connection
+def init_sqlite():
+    conn = sqlite3.connect("feedback.db")
+    cursor = conn.cursor()
+    # Create table if it doesn't exist
+    cursor.execute("""
+        CREATE TABLE IF NOT EXISTS Feedback (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            helpful TEXT NOT NULL,
+            question_type TEXT NOT NULL
+        )
+    """)
+    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
@@ -82,7 +90,10 @@ 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, "question_type": question_type})
+            cursor = conn.cursor()
+            cursor.execute("INSERT INTO Feedback (helpful, question_type) VALUES (?, ?)",
+                           (st.session_state.helpful, question_type))
+            conn.commit()
             st.success("Feedback submitted successfully.")
         except Exception as e:
             st.error(f"Failed to submit feedback: {e}")
@@ -102,4 +113,4 @@ st.markdown(
     </style>
     """,
     unsafe_allow_html=True
-)
+)