Update app.py

app.py

@@ -35,8 +35,111 @@ if "max_optimizer_iterations" not in st.session_state:
 
 # Navigation sidebar setup
 st.sidebar.title("JEE Roadmap Planner")
-page = st.sidebar.radio("Navigation", ["Home", "Roadmap Manager", "Task Analysis"])
+page = st.sidebar.radio("Navigation", ["Home", "Roadmap Manager", "Task Analysis","Roadmap Chatbot"])
 
+
+# For roadmap chatbot
+import sqlite3
+
+# Function to convert NL query to SQL
+def generate_sql_from_nl(prompt):
+    client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+
+    table_struct = """
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        date TEXT,
+        subject TEXT,
+        chapter_name TEXT,
+        task_type TEXT,
+        time TEXT,
+        subtopic TEXT
+    """
+
+    response = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=[
+            {"role": "system", "content": "You are an expert at converting natural language to SQL."},
+            {"role": "user", "content": f"""Convert this to an SQL query for the 'roadmap' table: {prompt}
+            Keep the table structure in mind: {table_struct}, 
+            Also dont include new line statements in the output of the query and just give me the query in the output.
+            Do not include ```sql and any other text."""}
+        ]
+    )
+    return response.choices[0].message.content.strip()
+
+# Function to convert SQL output to natural language
+def generate_nl_from_sql_output(prompt):
+    client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+
+    response = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=[
+            {"role": "system", "content": """You are a helpful chatbot for JEE roadmap analysis.
+            You will be given SQL-fetched roadmap data and must give a natural and interactive explanation to the user."""},
+            {"role": "user", "content": f"""Convert this data into interactive output: {prompt}"""}
+        ]
+    )
+    return response.choices[0].message.content.strip()
+
+# Function to fetch data from SQLite
+def fetch_data_from_sql(sql_query):
+    conn = sqlite3.connect("jee_roadmap.db")
+    cursor = conn.cursor()
+    cursor.execute(sql_query)
+    rows = cursor.fetchall()
+    conn.close()
+    return rows
+
+# Main function for chatbot
+def answer_user_query(prompt):
+    initialize_roadmap_db()
+    sql = generate_sql_from_nl(prompt)
+    rows = fetch_data_from_sql(sql)
+    return generate_nl_from_sql_output(rows)
+    
+def initialize_roadmap_db():
+    if not os.path.exists("jee_roadmap.db"):
+        try:
+            with open("full_roadmap.json") as f:
+                roadmap_data = json.load(f)
+
+            conn = sqlite3.connect("jee_roadmap.db")
+            cursor = conn.cursor()
+
+            cursor.execute("""
+            CREATE TABLE IF NOT EXISTS roadmap (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                date TEXT,
+                subject TEXT,
+                chapter_name TEXT,
+                task_type TEXT,
+                time TEXT,
+                subtopic TEXT
+            )
+            """)
+
+            for day in roadmap_data["schedule"]:
+                date = day["date"]
+                for subj in day["subjects"]:
+                    subject = subj["name"]
+                    for task in subj["tasks"]:
+                        cursor.execute("""
+                            INSERT INTO roadmap (date, subject, chapter_name, task_type, time, subtopic)
+                            VALUES (?, ?, ?, ?, ?, ?)
+                        """, (
+                            date,
+                            subject,
+                            task["ChapterName"],
+                            task["type"],
+                            task["time"],
+                            task["subtopic"]
+                        ))
+
+            conn.commit()
+            conn.close()
+            print("✅ Database created and data inserted successfully.")
+        except Exception as e:
+            print(f"⚠️ Error initializing database: {e}")
 # Function to load initial data
 def load_initial_data():
     with st.spinner("Loading roadmap data..."):
@@ -578,4 +681,17 @@ elif page == "Task Analysis":
                 
                 with col2:
                     st.subheader("Task Type Distribution")
-                    st.bar_chart(type_counts)
+                    st.bar_chart(type_counts)
+# ---- ROADMAP CHATBOT PAGE ----
+elif page == "Roadmap Chatbot":
+    st.title("🤖 Roadmap Chatbot Assistant")
+
+    user_query = st.text_input("Ask a question about your roadmap:", placeholder="e.g., What are my tasks on 14 Feb 2025?")
+
+    if st.button("Ask") and user_query:
+        with st.spinner("Thinking..."):
+            try:
+                response = answer_user_query(user_query)
+                st.markdown(response)
+            except Exception as e:
+                st.error(f"Error: {e}")