pr11

SessionState.py

@@ -0,0 +1,117 @@
+"""Hack to add per-session state to Streamlit.
+
+Usage
+-----
+
+>>> import SessionState
+>>>
+>>> session_state = SessionState.get(user_name='', favorite_color='black')
+>>> session_state.user_name
+''
+>>> session_state.user_name = 'Mary'
+>>> session_state.favorite_color
+'black'
+
+Since you set user_name above, next time your script runs this will be the
+result:
+>>> session_state = get(user_name='', favorite_color='black')
+>>> session_state.user_name
+'Mary'
+
+"""
+try:
+    import streamlit.ReportThread as ReportThread
+    from streamlit.server.Server import Server
+except Exception:
+    # Streamlit >= 0.65.0
+    import streamlit.report_thread as ReportThread
+    from streamlit.server.server import Server
+
+
+class SessionState(object):
+    def __init__(self, **kwargs):
+        """A new SessionState object.
+
+        Parameters
+        ----------
+        **kwargs : any
+            Default values for the session state.
+
+        Example
+        -------
+        >>> session_state = SessionState(user_name='', favorite_color='black')
+        >>> session_state.user_name = 'Mary'
+        ''
+        >>> session_state.favorite_color
+        'black'
+
+        """
+        for key, val in kwargs.items():
+            setattr(self, key, val)
+
+
+def get(**kwargs):
+    """Gets a SessionState object for the current session.
+
+    Creates a new object if necessary.
+
+    Parameters
+    ----------
+    **kwargs : any
+        Default values you want to add to the session state, if we're creating a
+        new one.
+
+    Example
+    -------
+    >>> session_state = get(user_name='', favorite_color='black')
+    >>> session_state.user_name
+    ''
+    >>> session_state.user_name = 'Mary'
+    >>> session_state.favorite_color
+    'black'
+
+    Since you set user_name above, next time your script runs this will be the
+    result:
+    >>> session_state = get(user_name='', favorite_color='black')
+    >>> session_state.user_name
+    'Mary'
+
+    """
+    # Hack to get the session object from Streamlit.
+
+    ctx = ReportThread.get_report_ctx()
+
+    this_session = None
+
+    current_server = Server.get_current()
+    if hasattr(current_server, '_session_infos'):
+        # Streamlit < 0.56
+        session_infos = Server.get_current()._session_infos.values()
+    else:
+        session_infos = Server.get_current()._session_info_by_id.values()
+
+    for session_info in session_infos:
+        s = session_info.session
+        if (
+            # Streamlit < 0.54.0
+            (hasattr(s, '_main_dg') and s._main_dg == ctx.main_dg)
+            or
+            # Streamlit >= 0.54.0
+            (not hasattr(s, '_main_dg') and s.enqueue == ctx.enqueue)
+            or
+            # Streamlit >= 0.65.2
+            (not hasattr(s, '_main_dg') and s._uploaded_file_mgr == ctx.uploaded_file_mgr)
+        ):
+            this_session = s
+
+    if this_session is None:
+        raise RuntimeError(
+            "Oh noes. Couldn't get your Streamlit Session object. "
+            'Are you doing something fancy with threads?')
+
+    # Got the session object! Now let's attach some state into it.
+
+    if not hasattr(this_session, '_custom_session_state'):
+        this_session._custom_session_state = SessionState(**kwargs)
+
+    return this_session._custom_session_state

app.py

@@ -1,58 +1,101 @@
+# import streamlit as st
+# import time
+
+# with st.form("my_input"):
+#    st.write("Input")
+#    # product
+#    product=st.text_input("product")
+#    # gender
+#    gender=st.radio("gender", ["male", "female"])
+#    # profession
+#    profession=st.text_input("profession")
+#    # hobby
+#    hobby=st.text_input("hobby")
+
+#    # Every form must have a submit button.
+#    col1, col2=st.columns(2)
+
+#    # Place a button in each column
+#    with col1:
+#         submitted = st.form_submit_button("Submit")
+
+#    with col2:
+#         clear = st.form_submit_button("Clear") 
+
+ 
+
+# with st.form("my_output"):
+#     if submitted:
+#        st.write("product", product)
+#        st.write("gender", gender)
+#        st.write("profession", profession)
+#        st.write("hobby", hobby)
+#     # Clear the user inputs
+#     if clear:
+#         st.experimental_rerun()
+
+# _LOREM_IPSUM = """
+# Lorem ipsum dolor sit amet, **consectetur adipiscing** elit, sed do eiusmod tempor
+# incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis
+# nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.
+# """
+
+# def stream_data():
+#     for word in _LOREM_IPSUM.split(" "):
+#         yield word + " "
+#         time.sleep(0.02)
+
+#     for word in _LOREM_IPSUM.split(" "):
+#         yield word + " "
+#         time.sleep(0.02)
+
+
+# if st.button("Stream data"):
+#     st.write_stream(stream_data)
+
+
 import streamlit as st
-from utils import generate_email
-
-    
-col1, col2=st.columns(2)
-with col1:
-    with st.container(height=500):
-        with st.form("my_input", clear_on_submit=True, border=False):
-            st.write("Input")
-            # product
-            product=st.text_input("product", key="product")
-            # gender
-            gender=st.radio("gender", ["male", "female", "non-binary"], key="gender")
-            # profession
-            profession=st.text_input("profession", key="profession")
-            # hobby
-            hobby=st.text_input("hobby", key="hobby")
-            # Every form must have a submit button.
-            submitted = st.form_submit_button(label='Submit')
-            # if submitted: 
-            #      response = generate_email(st.session_state.product, st.session_state.gender, st.session_state.profession, st.session_state.hobby)
-            #      st.session_state.email=response
-                 
-with col2:
-        if submitted:
-            with st.spinner('Wait for it...'):
-                response = generate_email(st.session_state.product, st.session_state.gender, st.session_state.profession, st.session_state.hobby)
-                st.session_state.email=response
-                print("st.session_state.email")
-                print(st.session_state.email)
-                with st.container(height=500):
-                    placeholder = st.empty()
-                    # Replace the chart with several elements:
-                    with placeholder.container():
-                        st.write("Product: " + st.session_state.product)
-                        st.write("Gender: " + st.session_state.gender) 
-                        st.write("Profession: " + st.session_state.profession) 
-                        st.write("Hobby: "+ st.session_state.hobby)
-                        st.write("Email: "+ st.session_state.email)
-
-if 'clicked' not in st.session_state:
-        st.session_state.clicked = False
-
-def click_button():
-    st.session_state.clicked = True
-
-st.button('Clear', on_click=click_button)
-
-if st.session_state.clicked:
-    # The message and nested widget will remain on the page
-    del st.session_state.product
-    del st.session_state.gender
-    del st.session_state.profession
-    del st.session_state.hobby
-    del st.session_state.email
-    del st.session_state.clicked
-    if submitted:
-        placeholder.empty()
+import SessionState
+
+# Initialize session state
+session_state = SessionState.get(product="", gender="", profession="", hobby="")
+
+# Create a form for user inputs
+with st.form("my_input"):
+    st.write("Input")
+    # product
+    product = st.text_input("product", value=session_state.product)
+    # gender
+    gender = st.radio("gender", ["male", "female"], index=["male", "female"].index(session_state.gender) if session_state.gender else 0)
+    # profession
+    profession = st.text_input("profession", value=session_state.profession)
+    # hobby
+    hobby = st.text_input("hobby", value=session_state.hobby)
+
+    # Every form must have a submit button.
+    submitted = st.form_submit_button("Submit")
+    clear = st.form_submit_button("Clear")
+
+# Display the user inputs
+with st.form("my_output"):
+    if submitted and not clear:
+        # Save inputs to session state
+        session_state.product = product
+        session_state.gender = gender
+        session_state.profession = profession
+        session_state.hobby = hobby
+
+        st.write("product", product)
+        st.write("gender", gender)
+        st.write("profession", profession)
+        st.write("hobby", hobby)
+
+    # Clear the user inputs
+    if clear:
+        # Clear session state
+        session_state.product = ""
+        session_state.gender = ""
+        session_state.profession = ""
+        session_state.hobby = ""
+
+        st.experimental_rerun()

requirements.txt

@@ -1,2 +0,0 @@
-torch
-transformers>=4.24.0

utils.py

@@ -1,51 +0,0 @@
-from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
-import re
-
-# Load saved model and tokenizer
-model_checkpoint = "24NLPGroupO/EmailGeneration"
-tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, truncation=True)
-model = AutoModelForCausalLM.from_pretrained(model_checkpoint)
-
-# Set up the generation pipeline
-generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
-
-def clean_generated_text(text):
-    #Basic cleaning
-    text = re.sub(r'^(Re:|Fwd:)', '', text)                         # Remove reply and forward marks
-    text = re.sub(r'Best regards,.*$', '', text, flags=re.DOTALL)   # Remove signature
-    text = re.sub(r'PHONE.*$', '', text, flags=re.DOTALL)           # Remove phone numbers
-    text = re.sub(r'Email:.*$', '', text, flags=re.DOTALL)          # Remove email addresses
-    text = re.sub(r'Cc:.*$', '', text, flags=re.DOTALL)             # Remove CC list
-    text = re.sub(r'\* Attachments:.*', '', text, flags=re.S)       # Remove Attachments
-    text = re.sub(r'©️ .*$', '', text, flags=re.DOTALL)              # Remove copyright and ownership statements
-    text = re.sub(r'URL', '', text)                                 # Remove URLs
-    text = re.sub(r'NUMBER', '10', text)                            # Replace 'NUMBER' with a real number
-    text = re.sub(r'CURRENCYNUMBER', 'USD 100', text)               # Replace 'CURRENCYNUMBER' with a real value
-    text = re.sub(r'About Us.*', '', text, flags=re.DOTALL)         # Remove 'About Us' and all following text
-    text = re.sub(r'\d+ [^\s]+ St\.?,?.*?\d{5}', '', text)          # Remove street
-    text = re.sub(r'\d+ [^\s]+ Ave\.?,?.*?\d{5}', '', text)         # Remove avenues
-    text = re.sub(r'\d+ [^\s]+ Rd\.?,?.*?\d{5}', '', text)          # Remove roads
-    text = re.sub(r'\d+ [^\s]+ Ln\.?,?.*?\d{5}', '', text)          # Remove lanes
-    text = re.sub(r'\d+ [^\s]+ Blvd\.?,?.*?\d{5}', '', text)        # Remove boulevards
-    text = re.sub(r'\d+ [^\s]+ Dr\.?,?.*?\d{5}', '', text)          # Remove drives
-    text = re.sub(r'\d+ [^\s]+ Ct\.?,?.*?\d{5}', '', text)          # Remove courts
-    return text.strip()
-
-def generate_email(product, gender, profession, hobby):
-    input_text = f"{product} {gender} {profession} {hobby}"
-    result = generator(
-        input_text,                # The starting text that guides the model on what to generate
-        max_length=256,            # Set a suitable maximum length
-        top_k=40,                  # Consider more top options words
-        top_p=0.6,                 # Control the probability range for word choices
-        temperature=0.4,           # Control the randomness of generation
-        repetition_penalty=1.5,    # Reduce content repetition
-        num_return_sequences=2,    # Generate three texts
-        do_sample=True
-    )       
-    # Clean each generated text
-    cleaned_texts = [clean_generated_text(seq['generated_text']) for seq in result]
-    # Choose the best text based on length and clarity
-    best_text = max(cleaned_texts, key=len)
-    return best_text
-