Update app.py

app.py

@@ -1,16 +1,84 @@
 import streamlit as st
-import torch
-from transformers import GPT2Tokenizer, GPT2Model
+import urllib
+import re
+import pandas as pd
+import random
 
-tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
-model = GPT2Model.from_pretrained('gpt2')
+# Set the page config
+st.set_page_config(
+    layout="centered", # or "centered"
+    initial_sidebar_state="auto", # or "expanded" or "collapsed"
+    page_title="WordGPT", # or any other title
+    page_icon="📝" # or any other emoji or image
+)
 
-st.title("GPT-2 Text Generation")
+# Give a large title to the page
+st.title("WordGPT")
 
-text = st.text_input("Enter the starting text:", "Replace me by any text you'd like.")
+# read the csv file
+df = pd.read_csv("ielts.csv")
 
-if st.button("Generate"):
-    encoded_input = tokenizer(text, return_tensors='pt', padding=False, truncation=False)
-    output = model(**encoded_input)
-    generated_text = tokenizer.decode(torch.argmax(output[0], dim=1), skip_special_tokens=True)
-    st.write(generated_text)
+# create two columns
+col1, col2 = st.columns(2)
+
+# get all the values from the first column of the dataframe as a list
+values = df.iloc[:,0].tolist()
+
+# join the values with newline characters as a string
+text = "\n".join(values)
+
+placeholder = col1.empty()
+
+# display the value in the textarea of the first column
+words = placeholder.text_area("Value from ielts.csv", text, height=200)
+
+# Create a button in col1 named "Shuffle the words"
+shuffle_button = col1.button("Shuffle the words")
+
+# Shuffle the words in the text area and display them in col2
+if shuffle_button:
+    words = words.split("\n")
+    random.shuffle(words)
+    words = "\n".join(words)
+    placeholder.empty()
+    words = placeholder.text_area("Shuffle the words", words, height=200)
+
+# Check if the user has entered any words
+if words:
+    # Split the words by any common symbol or line break and strip any whitespace
+    words = [word.strip() for word in re.split("[,.;:!?]|\\n", words)]
+
+    # Select only the first 12 words of the list
+    words = " ".join(re.findall(r"\w+", " ".join(words))[:12])
+
+    # Create a Bing query in the desired format
+    query = f"write a short story with these words involved: {words}"
+    query_origin = query
+
+    # Replace all spaces with "+" symbols
+    # query = query.replace(" ", "+")
+
+    # Encode the query for the URL
+    query = urllib.parse.quote(query)
+
+    # Create a Bing URL with the query
+    url = f"https://www.bing.com/search?q={query}"
+
+    # Put the rest of the elements in the right column
+    col2.write(f"**Your Bing query is:** {query_origin}")
+    col2.markdown(f'<a href="{url}" target="_blank"><button>Let Bing AI create a story</button></a>', unsafe_allow_html=True)
+
+    # Create a form in the right column with the clear_on_submit parameter set to True
+    with col2.form("story_form", clear_on_submit=True):
+        # Put a textarea in the form
+        story = st.text_area("Enter your own story or edit the Bing AI story", value="", height=200)
+
+        # Put a submit button in the form
+        submitted = st.form_submit_button("Submit")
+
+        # Check if the user has submitted the form
+        if submitted:
+            # Open the file in append mode
+            with open("my_story.txt", "a") as f:
+                # Write the story to the file
+                f.write(story + "\n")