Update app.py

app.py

@@ -1,4 +1,3 @@
-# Import necessary libraries
 import streamlit as st
 from langchain_community.document_loaders import PyPDFLoader
 import openai
@@ -7,10 +6,11 @@ from langchain_core.output_parsers import StrOutputParser
 from langchain.chat_models import ChatOpenAI
 from fpdf import FPDF
 import os
+import pandas as pd
 
 # Set up Streamlit UI
 st.title('Educational Assistant')
-st.header('Summary, Quiz Generator, and Q&A')
+st.header('Summary, Quiz Generator, Q&A, and Study Plan')
 st.sidebar.title('Drop your PDF here')
 
 # Input OpenAI API key from keyboard
@@ -18,8 +18,8 @@ openai_api_key = st.sidebar.text_input("Enter your OpenAI API Key", type="passwo
 
 user_file_upload = st.sidebar.file_uploader(label='', type='pdf')
 
-# Sidebar option selection for Summary, Quiz, or Q&A
-option = st.sidebar.radio("Choose an option", ('Generate Summary', 'Generate Quiz', 'Ask a Question'))
+# Sidebar option selection for Summary, Quiz, Q&A, or Study Plan
+option = st.sidebar.radio("Choose an option", ('Generate Summary', 'Generate Quiz', 'Ask a Question', 'Generate Study Plan'))
 
 # Input for asking questions (only visible when "Ask a Question" is selected)
 question_input = None
@@ -99,6 +99,19 @@ if openai_api_key:
         output_parser = StrOutputParser()
         chain_3 = prompt_3 | llm_qa | output_parser
 
+        ## Prompt Template for Study Plan
+        prompt_4 = ChatPromptTemplate.from_messages(
+            [
+                ("system", "You are a smart assistant. Based on the content of the user's PDF, generate a 7-day study plan. Divide the content into 7 topics and assign each topic to a day. Please make it logical and balanced."),
+                ("user", "{data}")
+            ]
+        )
+
+        # Pass the OpenAI API key explicitly to the ChatOpenAI instance
+        llm_study_plan = ChatOpenAI(model="gpt-4o-mini", openai_api_key=openai_api_key)  # Pass the key here
+        output_parser = StrOutputParser()
+        chain_4 = prompt_4 | llm_study_plan | output_parser
+
     if option == 'Generate Summary':
         # Generate summary
         summary_response = chain_1.invoke({'data': data})
@@ -129,5 +142,27 @@ if openai_api_key:
             # Generate PDF for the question answer and offer it as a download
             pdf_filename = generate_pdf(question_answer_response, filename="question_answer_response.pdf")
             st.download_button("Download Answer as PDF", data=open(pdf_filename, "rb").read(), file_name=pdf_filename, mime="application/pdf")
+
+    elif option == 'Generate Study Plan':
+        # Generate study plan
+        study_plan_response = chain_4.invoke({'data': data})
+        st.write(study_plan_response)
+        
+        # Extract the study plan and convert it into a structured format
+        study_plan = study_plan_response.strip().split("\n")
+        
+        # Assuming the study plan is a list of 7 days with topics
+        days = ["Day 1", "Day 2", "Day 3", "Day 4", "Day 5", "Day 6", "Day 7"]
+        topics = [plan.split(":")[1].strip() if ":" in plan else "" for plan in study_plan]
+
+        # Create a DataFrame to display the study plan in tabular form
+        study_plan_df = pd.DataFrame(list(zip(days, topics)), columns=["Day", "Topics to Study"])
+        
+        st.table(study_plan_df)
+
+        # Generate PDF for the study plan and offer it as a download
+        pdf_filename = generate_pdf(study_plan_response, filename="study_plan_response.pdf")
+        st.download_button("Download Study Plan as PDF", data=open(pdf_filename, "rb").read(), file_name=pdf_filename, mime="application/pdf")
+
 else:
-    st.sidebar.warning("Please enter your OpenAI API Key to proceed.")
+    st.sidebar.warning("Please enter your OpenAI API Key to proceed.")