Link Caching and Moving Download to Sidebar

app.py

@@ -20,7 +20,6 @@ nlp = spacy.load("en_core_web_sm")
 user_agent = 'QGen/1.0 ([email protected])'
 wiki_wiki = wikipediaapi.Wikipedia(user_agent= user_agent,language='en')
 
-
 model = None
 tokenizer = None
 def load_model():
@@ -71,6 +70,7 @@ def map_keywords_to_sentences(text, keywords, context_window_size):
     return keyword_sentence_mapping
 
 # Function to perform entity linking using Wikipedia API
+@lru_cache(maxsize=128)
 def entity_linking(keyword):
     page = wiki_wiki.page(keyword)
     if page.exists():
@@ -78,7 +78,7 @@ def entity_linking(keyword):
     return None
 
 # Function to generate questions using beam search
-def generate_question(context, answer, num_beams=5):
+def generate_question(context, answer, num_beams):
     input_text = f"<context> {context} <answer> {answer}"
     input_ids = tokenizer.encode(input_text, return_tensors='pt')
     outputs = model.generate(input_ids, num_beams=num_beams, early_stopping=True)
@@ -140,21 +140,13 @@ if st.button("Generate Questions"):
         
         # Export buttons
         if data is not None:
-        # if st.button("Export to CSV"):
-        # if downlaod_csv:
-        #     csv_data = export_to_csv(data)
-        #     st.success("Questions exported to questions.csv")
-        #     st.download_button(label="Download CSV", data=csv_data, file_name='questions.csv', mime='text/csv')
-            csv_data = export_to_csv(data)
-            st.download_button(label="Download CSV", data=csv_data, file_name='questions.csv', mime='text/csv')
+            with st.sidebar:
+                st.subheader('Download Content')
+                csv_data = export_to_csv(data)
+                st.download_button(label="CSV Format", data=csv_data, file_name='questions.csv', mime='text/csv')
 
-        # if st.button("Export to PDF"):
-        # if download_pdf:
-        #     pdf_data = export_to_pdf(data)
-        #     st.success("Questions exported to questions.pdf")
-        #     st.download_button(label="Download PDF", data=pdf_data, file_name='questions.pdf', mime='application/pdf')
-            pdf_data = export_to_pdf(data)
-            st.download_button(label="Download PDF", data=pdf_data, file_name='questions.pdf', mime='application/pdf')
+                pdf_data = export_to_pdf(data)
+                st.download_button(label="PDF Format", data=pdf_data, file_name='questions.pdf', mime='application/pdf')
 
     else:
         st.write("Please enter some text to generate questions.")