Update app.py

app.py

@@ -34,32 +34,30 @@ def clean_cache():
         if os.path.isfile(file_path) and now - os.path.getmtime(file_path) > 900:  # 15 minutes
             os.remove(file_path)
             
-def convert_image_ffmpeg(image, target_format):
+def convert_image(image_path, target_format):
     try:
-        # Create a temporary directory for the uploaded file
-        temp_dir = tempfile.mkdtemp()
+        # Pastikan target format valid
+        if target_format.lower() not in supported_formats:
+            return "Error: Unsupported target format."
 
-        # Sanitize the filename and save the uploaded video to the temp directory
-        base_name = os.path.splitext(os.path.basename(image.name))[0]
-        sanitized_base_name = sanitize_filename(base_name)
-        image_path = os.path.join(temp_dir, f"{sanitized_base_name}.png")  # Saving as mp4 by default for now
+        # Sanitize file path
+        sanitized_base_name = sanitize_filename(os.path.splitext(os.path.basename(image_path))[0])
         
-        # Nama file keluaran dengan awalan yang diinginkan
-        output_file = f"flowly_ai_image_converter_{sanitized_base_name}.{target_format.lower()}"
+        # Nama file keluaran
+        output_file = os.path.join(CACHE_DIR, f"flowly_ai_image_converter_{sanitized_base_name}.{target_format.lower()}")
 
-        with open(image_path, "wb") as f:
-            f.write(image.getbuffer())  # Save the uploaded video to a local file
-            
         # Gunakan ffmpeg untuk konversi
-        ffmpeg.input(image_path).output(output_file, format=target_format).overwrite_output().run(capture_stdout=True, capture_stderr=True)
+        ffmpeg.input(image_path).output(output_file, format=target_format).overwrite_output().run(
+            capture_stdout=True, capture_stderr=True
+        )
 
-        return output_file
+        return output_file  # Return path ke file hasil
     except Exception as e:
         return f"Error: {e}"
-
+        
 # Antarmuka Gradio
 interface = gr.Interface(
-    fn=convert_image_ffmpeg,
+    fn=convert_image,
     inputs=[
         gr.Image(label="Upload Image", type="filepath", height=512),  # Input gambar
         gr.Dropdown(label="Select Target Format", choices=supported_formats)  # Pilih format