Update app.py

app.py

@@ -51,9 +51,8 @@ def text2story(text):
         generator = pipeline("text-generation", model="2173ars/llama-3-8b-Instruct-bnb-4bit-personal-shortstory")
         
         # Create a prompt for the story
-        prompt = f"{text}"
+        prompt = f" Story Context : {text} ; The story should have: - A clear beginning, middle, and end - Well-developed characters with motivation - A central theme or message - Engaging descriptions - Natural dialogue (if appropriate) - An emotional or thought-provoking conclusion "
         
-        # Generate the story - limit to a smaller max_length due to model size
         story = generator(prompt,
                          min_length=100,
                          max_length=800,  
@@ -78,7 +77,7 @@ def text2story(text):
     except Exception as e:
         # Fallback to simpler model if the advanced one fails
         fallback_generator = pipeline('text-generation', model='gpt2')
-        fallback_prompt = f"Create a short story about this scene: {text}\n\nStory:"
+        fallback_prompt = f" Story Context : {text} ; The story should have: - A clear beginning, middle, and end - Well-developed characters with motivation - A central theme or message - Engaging descriptions - Natural dialogue (if appropriate) - An emotional or thought-provoking conclusion "
         fallback_story = fallback_generator(fallback_prompt, min_length=100, num_return_sequences=1)[0]['generated_text']
         return fallback_story.replace(fallback_prompt, "").strip()
 
@@ -155,18 +154,17 @@ progress_placeholder = st.empty()
 
 # File uploader with Cantonese
 with st.container():
-    st.subheader("請選擇一張畫作上載，格式必須係 JPG、JPEG 或 PNG！ 最大 200 MB !")
-    uploaded_file = st.file_uploader(
-        "",
-        type=["jpg", "jpeg", "png"],  # Limit file types
-        key="upload"
-    )
-    
     if uploaded_file is not None:
         st.success("上載成功！")
     else:
         st.info("請選擇一張畫作上載，格式必須係 JPG、JPEG 或 PNG！ 最大 200 MB !")
         
+    uploaded_file = st.file_uploader(
+        "",
+        type=["jpg", "jpeg", "png"],  # Limit file types
+        key="upload"
+    )
+ 
 if uploaded_file is not None:
     # Save uploaded file
     bytes_data = uploaded_file.getvalue()