using session_id for all different users

app.py

@@ -15,7 +15,6 @@ from nltk.tokenize import sent_tokenize
 nltk.download('wordnet')
 from nltk.corpus import wordnet
 import random
-from sense2vec import Sense2Vec
 import sense2vec
 from wordcloud import WordCloud
 import matplotlib.pyplot as plt
@@ -27,6 +26,7 @@ from spellchecker import SpellChecker
 from transformers import pipeline
 import re
 import pymupdf
+import uuid
 print("***************************************************************")
 
 st.set_page_config(
@@ -41,6 +41,27 @@ st.set_page_config(
 user_agent = 'QGen/1.0 ([email protected])'
 wiki_wiki = wikipediaapi.Wikipedia(user_agent= user_agent,language='en')
 
+def get_session_id():
+    if 'session_id' not in st.session_state:
+        st.session_state.session_id = str(uuid.uuid4())
+    return st.session_state.session_id
+
+def initialize_state(session_id):
+    if 'session_states' not in st.session_state:
+        st.session_state.session_states = {}
+
+    if session_id not in st.session_state.session_states:
+        st.session_state.session_states[session_id] = {
+            'generated_questions': [],
+            # add other state variables as needed
+        }
+    return st.session_state.session_states[session_id]
+
+def get_state(session_id):
+    return st.session_state.session_states[session_id]
+
+def set_state(session_id, key, value):
+    st.session_state.session_states[session_id][key] = value
 
 @st.cache_resource
 def load_model():
@@ -314,7 +335,8 @@ def assess_question_quality(context, question, answer):
 def main():
     # Streamlit interface
     st.title(":blue[Question Generator System]")
-
+    session_id = get_session_id()
+    state = initialize_state(session_id)
     # Initialize session state
     if 'generated_questions' not in st.session_state:
         st.session_state.generated_questions = []
@@ -349,7 +371,8 @@ def main():
         segments = segment_text(text)
     generate_questions_button = st.button("Generate Questions")
     if generate_questions_button and text:
-        st.session_state.generated_questions = []
+        state['generated_questions'] = []
+        # st.session_state.generated_questions = []
         for text in segments:
             keywords = extract_keywords(text, extract_all_keywords)
             print(f"\n\nFinal Keywords in Main Function: {keywords}\n\n")
@@ -372,12 +395,17 @@ def main():
                     "complexity_score" : complexity_score,
                     "spelling_correctness" : spelling_correctness,
                 }
-                st.session_state.generated_questions.append(tpl)
+                # st.session_state.generated_questions.append(tpl)
+                state['generated_questions'].append(tpl)
+     
+        set_state(session_id, 'generated_questions', state['generated_questions'])
     
     # sort question based on their quality score
-    st.session_state.generated_questions = sorted(st.session_state.generated_questions,key = lambda x: x['overall_score'], reverse=True)
+    # st.session_state.generated_questions = sorted(st.session_state.generated_questions,key = lambda x: x['overall_score'], reverse=True)
+    state['generated_questions'] = sorted(state['generated_questions'],key = lambda x: x['overall_score'], reverse=True)
     # Display generated questions
-    if st.session_state.generated_questions:
+    # if st.session_state.generated_questions:
+    if state['generated_questions']:
         st.header("Generated Questions:",divider='blue')
         for i, q in enumerate(st.session_state.generated_questions):
             # with st.expander(f"Question {i+1}"):
@@ -396,10 +424,11 @@ def main():
                 if linked_entity:
                     st.write(f"**Entity Link:** {linked_entity}")
             if show_qa_scores is True:
-                st.write(f"**Overall Quality Score:** {q['overall_score']:.2f}")
-                st.write(f"**Relevance Score:** {q['relevance_score']:.2f}")
-                st.write(f"**Complexity Score:** {q['complexity_score']:.2f}")
-                st.write(f"**Spelling Correctness:** {q['spelling_correctness']:.2f}")
+                m1,m2,m3,m4 = st.columns([1.7,1,1,1])
+                m1.metric("Overall Quality Score", value=f"{q['overall_score']:,.2f}")
+                m2.metric("Relevance Score", value=f"{q['relevance_score']:,.2f}")
+                m3.metric("Complexity Score", value=f"{q['complexity_score']:,.2f}")
+                m4.metric("Spelling Correctness", value=f"{q['spelling_correctness']:,.2f}")
 
             # q['context'] = st.text_area(f"Edit Context {i+1}:", value=q['context'], key=f"context_{i}")
             if enable_feedback_mode:
@@ -411,7 +440,8 @@ def main():
             st.write("---")
 
         # Export buttons
-        if st.session_state.generated_questions:
+        # if st.session_state.generated_questions:
+        if state['generated_questions']:
             with st.sidebar:
                 csv_data = export_to_csv(st.session_state.generated_questions)
                 st.download_button(label="Download CSV", data=csv_data, file_name='questions.csv', mime='text/csv')