Update app.py

app.py

@@ -18,14 +18,13 @@ openai_api_key = st.sidebar.text_input("Enter your OpenAI API Key", type="passwo
 
 user_file_upload = st.sidebar.file_uploader(label='', type='pdf')
 
-summary_clicked = st.button('Generate Summary')
-quiz_clicked = st.button('Generate Quiz')
+# Sidebar option selection for Summary, Quiz, or Q&A
+option = st.sidebar.radio("Choose an option", ('Generate Summary', 'Generate Quiz', 'Ask a Question'))
 
-# Input for asking questions
-question_input = st.text_input("Enter your question about the document:")
-
-# Button to trigger question answering, placed after the input
-ask_question_clicked = st.button('Ask a Question')
+# Input for asking questions (only visible when "Ask a Question" is selected)
+question_input = None
+if option == 'Ask a Question':
+    question_input = st.text_input("Enter your question about the document:")
 
 # Function to generate a PDF and allow download
 def generate_pdf(response, filename="response.pdf"):
@@ -100,7 +99,7 @@ if openai_api_key:
         output_parser = StrOutputParser()
         chain_3 = prompt_3 | llm_qa | output_parser
 
-    if summary_clicked:
+    if option == 'Generate Summary':
         # Generate summary
         summary_response = chain_1.invoke({'data': data})
         st.write(summary_response)
@@ -109,7 +108,7 @@ if openai_api_key:
         pdf_filename = generate_pdf(summary_response, filename="summary_response.pdf")
         st.download_button("Download Summary as PDF", data=open(pdf_filename, "rb").read(), file_name=pdf_filename, mime="application/pdf")
         
-    elif quiz_clicked:
+    elif option == 'Generate Quiz':
         # Generate quiz
         quiz_response = chain_2.invoke({'data': data})
         st.write(quiz_response)
@@ -118,13 +117,17 @@ if openai_api_key:
         pdf_filename = generate_pdf(quiz_response, filename="quiz_response.pdf")
         st.download_button("Download Quiz as PDF", data=open(pdf_filename, "rb").read(), file_name=pdf_filename, mime="application/pdf")
         
-    elif ask_question_clicked and question_input:
-        # Generate answer for the user's question
-        question_answer_response = chain_3.invoke({'data': data, 'question': question_input})
-        st.write(question_answer_response)
+    elif option == 'Ask a Question' and question_input:
+        # Add a "Generate Answer" button
+        generate_answer = st.button("Generate Answer")
         
-        # 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")
+        if generate_answer:
+            # Generate answer for the user's question
+            question_answer_response = chain_3.invoke({'data': data, 'question': question_input})
+            st.write(question_answer_response)
+            
+            # 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")
 else:
     st.sidebar.warning("Please enter your OpenAI API Key to proceed.")