|
|
import streamlit as st |
|
|
import openai |
|
|
|
|
|
|
|
|
openai.api_key = st.secrets["OPENAI_API_KEY"] |
|
|
|
|
|
|
|
|
st.title("Welcome to JoyStory! 🎉") |
|
|
st.subheader("Let's create fun and imaginative stories together!") |
|
|
|
|
|
|
|
|
level = st.selectbox("Choose your level:", ["Beginner", "Intermediate", "Advanced"]) |
|
|
|
|
|
|
|
|
start_button = st.button("Start Creating Your Story!") |
|
|
|
|
|
|
|
|
|
|
|
if start_button: |
|
|
st.session_state["level"] = level |
|
|
st.session_state["story_started"] = True |
|
|
|
|
|
if st.session_state.get("story_started"): |
|
|
st.experimental_rerun() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if st.session_state.get("story_started"): |
|
|
st.header("JoyStory - Let's Create!") |
|
|
|
|
|
|
|
|
if "story_text" not in st.session_state: |
|
|
st.session_state["story_text"] = "Once upon a time, in a magical forest..." |
|
|
|
|
|
st.write(st.session_state["story_text"]) |
|
|
|
|
|
|
|
|
user_input = st.text_input("Add your sentence:") |
|
|
if st.button("Submit"): |
|
|
|
|
|
response = openai.Completion.create( |
|
|
model="text-davinci-003", |
|
|
prompt=st.session_state["story_text"] + " " + user_input, |
|
|
max_tokens=50 |
|
|
) |
|
|
ai_text = response.choices[0].text.strip() |
|
|
|
|
|
|
|
|
st.session_state["story_text"] += " " + user_input + " " + ai_text |
|
|
st.experimental_rerun() |
|
|
|
|
|
|
