Update app.py

app.py

@@ -1,38 +1,31 @@
 import streamlit as st
 import io
-from PIL import Image
-import easyocr
 from transformers import pipeline, set_seed
 
-# Create a text2text-generation pipeline with the "t5-base" model
-pipe = pipeline("text2text-generation", model="t5-base")
+# Create a text2text-generation pipeline with the "google/flan-t5-base" model
+pipe = pipeline("text2text-generation", model="google/flan-t5-base")
 
 # Initialize the EasyOCR reader for text extraction from images
-ocr_reader = easyocr.Reader(['en'])
-
-st.title("Text-to-Story Generator")
-
-uploaded_file = st.file_uploader("Upload an image:")
-
-if uploaded_file is not None:
-    # Read the uploaded image
-    image = Image.open(uploaded_file)
-
-    # Extract text from the image using OCR
-    ocr_results = ocr_reader.readtext(image)
-    extracted_text = " ".join([res[1] for res in ocr_results])
-    st.markdown("**Extracted text:**")
-    st.markdown(extracted_text)
-
-    if extracted_text:
-        # Use the pipeline to generate a story based on the extracted text
-        story_prompt = f"Once upon a time, in a world where '{extracted_text}',"
-        set_seed(42)  # Set a seed for reproducibility
-        story = pipe(story_prompt, max_length=300, do_sample=True)[0]["generated_text"]
-        st.markdown("**Story:**")
-        st.markdown(story)
-    else:
-        st.warning("No text extracted from the image.")
-
-else:
-    st.markdown("Please upload an image to extract text and generate a story.")
+# ocr_reader = easyocr.Reader(['en'])
+
+# Remove the text extraction code
+
+# Create input fields for the title, character, era, genre, and ending
+title = st.text_input("Title:")
+character = st.text_input("Character (hero):")
+era = st.text_input("Era of story:")
+genre = st.selectbox("Genre:", ["Action", "Novelistic", "Love"])
+ending = st.selectbox("Ending:", ["Happy", "Sad"])
+
+# Generate a story prompt based on the user input
+story_prompt = f"Once upon a time, in the {era}, there lived a {character} named {character}. {character} was a {genre} hero/heroine, and one day, they went on a quest to..."
+
+# Set a seed for reproducibility
+set_seed(42)
+
+# Generate the story
+story = pipe(story_prompt, max_length=300, do_sample=True)[0]["generated_text"]
+
+# Display the story to the user
+st.markdown("**Story:**")
+st.markdown(story)