Update app.py

app.py

@@ -5,12 +5,14 @@ import google.generativeai as genai
 from langchain_google_genai import ChatGoogleGenerativeAI
 from dotenv import load_dotenv
 import streamlit as st
-st.set_page_config(layout="wide", page_title="QA Pair Generation from Documents",page_icon='deep-learning.png')
+
+st.set_page_config(layout="wide", page_title="QA Pair Generation from Documents", page_icon='deep-learning.png')
 
 temperature = 0.3
 pages = []
 numPairs = 2
 option = ''
+FileName= ''
 optionCategory = ("Long QA Pairs", "MCQs", "Short QA Pairs")
 
 load_dotenv()
@@ -20,18 +22,25 @@ model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=temperature)
 def LongQAPairs():
     response = []
     with st.spinner('Generating Long Question Answer Pairs...'):
-        response =getLongQAPairs(pages[0:len(pages) - 1], numPairs, model)
+        response = getLongQAPairs(pages[0:len(pages) - 1], numPairs, model)
+
+    if len(FileName) != 5:
+        downloadFile(response,FileName)
 
     for qaPair in response:
         with st.chat_message("user"):
             st.write("Question : {}".format(qaPair['question']))
             st.write("Answer : {}".format(qaPair['answer']))
 
+
 def ShortQAPairs():
     response = []
     with st.spinner('Generating Short Question Answer Pairs...'):
         response = getShortQAPairs(pages[0:len(pages) - 1], numPairs, model)
 
+    if len(FileName) !=5:
+        downloadFile(response, FileName)
+
     for qaPair in response:
         with st.chat_message("user"):
             st.write("Question : {}".format(qaPair['question']))
@@ -43,9 +52,13 @@ def McqQAPairs():
     with st.spinner('Generating MCQ Question Answer Pairs...'):
         response = getMcqQAPairs(pages[0:len(pages) - 1], numPairs, model)
 
+    if len(FileName) != 5:
+        downloadFile(response,FileName)
+
     for qaPair in response:
         with st.chat_message("user"):
-            st.radio(label=qaPair['question'],options=qaPair["options"],disabled=True,index=qaPair['correct_option_index'])
+            st.radio(label=qaPair['question'], options=qaPair["options"], disabled=True,
+                     index=qaPair['correct_option_index'])
 
 
 with st.sidebar:
@@ -88,7 +101,12 @@ with st.container(border=True):
             index=None,
             placeholder="Select Category of Question Answer Pairs",
         )
-        numPairs = st.number_input('Number of QA Pairs', min_value=1, max_value=20, step=2,value=2)
+        numPairs = st.number_input('Number of QA Pairs', min_value=1, max_value=20, step=1, value=2)
+        with st.expander("Download your File Now"):
+            with st.container(border=True):
+                FileName = st.text_input('Enter Your File Name',
+                                     placeholder=' eg. Data, QA-Chap1')
+                FileName = FileName + '.json'
 
 if st.button("Generate", type="primary"):
     if option == "Long QA Pairs" and len(pages) and option in optionCategory: