Update app.py

app.py

@@ -61,7 +61,7 @@ def analyze_image(base64_image):
             },
         ]
         + generate_new_line(base64_image),
-        max_tokens=500,
+        max_tokens=100,
     )
     response_text = response.choices[0].message.content
     return response_text
@@ -82,8 +82,22 @@ def save_uploaded_file(uploaded_file):
     
     return os.path.join(save_path, 'temp')
 
-def process():
-    pass # Assuming 'process' is defined in 'your_processing_module'
+
+def save_audio_file(audio):
+    # Create a directory to save the file
+    save_path = 'audio'
+    if not os.path.exists(save_path):
+        os.makedirs(save_path)
+
+    # Save the file
+    with open(os.path.join(save_path, 'temp.mp3'), "wb") as f:
+        f.write(audio)
+    
+    return os.path.join(save_path, 'temp')
+
+
+from moviepy.editor import ImageClip, AudioFileClip
+
 
 def main():
     st.title("Image to Audio Converter")
@@ -103,6 +117,30 @@ def main():
         if audio_file is not None:
             st.audio(audio_file, format='audio/mp3')
             st.download_button('Download Audio', audio_file, file_name='narrated.mp3')
+            audio_filename = 'narrated.mp3'
+            with open(audio_filename, 'wb') as f:
+                f.write(audio_file)
+
+            print('creating video')
+            # Create a video clip from the static image
+            video_clip = ImageClip(path).set_duration(AudioFileClip(audio_filename).duration)
+
+
+            # Set the audio of the video clip as the generated audio file
+            video_clip = video_clip.set_audio(AudioFileClip(audio_filename))
+
+            print('video created')
+
+            # Specify the filename for the video
+            video_filename = "narrated_video.mp4"
+
+            # Write the video file to disk
+            video_clip.write_videofile(video_filename, codec="libx264", audio_codec="aac", fps=24)
+
+            # Provide a download button for the video
+            with open(video_filename, "rb") as file:
+                st.video(video_filename)
+                st.download_button('Download Video', file, file_name=video_filename)
 
 if __name__ == "__main__":
     main()