Update app.py

app.py

@@ -57,31 +57,28 @@ st.header("Turn Your Image to Audio Story")
 uploaded_file = st.file_uploader("Select an Image...")
 
 if uploaded_file is not None:
-    # 保存上传的文件到临时文件
-    temp_file_path = "temp_image.jpg"
-    bytes_data = uploaded_file.getvalue()
-    with open(temp_file_path, "wb") as file:
-        file.write(bytes_data)
+    print(uploaded_file)
+    # 使用临时文件处理上传的图像
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as temp_file:
+        temp_file.write(uploaded_file.getvalue())
+        temp_file_path = temp_file.name
+
     st.image(uploaded_file, caption="Uploaded Image",
-             use_container_width=True)
+             use_container_width=True)  # 修改为 use_container_width
 
-    # Stage 1: Image to Text
+    #Stage 1: Image to Text
     st.text('Processing img2text...')
     scenario = img2text(temp_file_path)
     if scenario:
         st.write(scenario)
 
-        # 删除临时文件
-        if os.path.exists(temp_file_path):
-            os.remove(temp_file_path)
-
-        # Stage 2: Text to Story
+        #Stage 2: Text to Story
         st.text('Generating a story...')
         story = text2story(scenario)
         if story:
             st.write(story)
 
-            # Stage 3: Story to Audio data
+            #Stage 3: Story to Audio data
             st.text('Generating audio data...')
             audio_data = text2audio(story)
             if audio_data:
@@ -89,4 +86,7 @@ if uploaded_file is not None:
                 if st.button("Play Audio"):
                     st.audio(audio_data,
                              format="audio/mpeg",
-                             start_time=0)
+                             start_time=0)
+
+    # 删除临时文件
+    os.remove(temp_file_path)