Update api.py

api.py

@@ -1,30 +1,36 @@
 # api.py
-from flask import Flask, request, jsonify
 import os
 import zipfile
 import rarfile
 from PyPDF2 import PdfReader, PdfWriter
-from moviepy.editor import VideoFileClip
 from pydub import AudioSegment
 from PIL import Image
 import pdf2image
-
-app = Flask(__name__)
+import streamlit as st
+import ffmpeg
 
 def convert_file(input_path, output_path):
     if input_path.endswith('.pdf'):
+        # Convert PDF to WEBP
         images = pdf2image.convert_from_path(input_path)
         for i, image in enumerate(images):
             image.save(f"{os.path.splitext(output_path)[0]}_page_{i+1}.webp", 'WEBP')
+
+        # Convert WEBP back to PDF
         webp_images = [Image.open(f"{os.path.splitext(output_path)[0]}_page_{i+1}.webp") for i in range(len(images))]
         webp_images[0].save(output_path, 'PDF', resolution=100.0, save_all=True, append_images=webp_images[1:])
+
     elif input_path.endswith('.mp4'):
-        video = VideoFileClip(input_path)
-        video.write_videofile(output_path, codec='libx264')
+        # Convert MP4 to MKV using ffmpeg
+        ffmpeg.input(input_path).output(output_path, codec='libx264').run()
+
     elif input_path.endswith('.wav'):
+        # Convert WAV to MP3
         audio = AudioSegment.from_wav(input_path)
         audio.export(output_path, format='mp3')
+
     elif input_path.endswith(('.png', '.jpg')):
+        # Convert PNG/JPG to WEBP
         image = Image.open(input_path)
         image.save(output_path, 'WEBP')
 
@@ -57,13 +63,58 @@ def get_output_extension(file):
         return '.webp'
     return ''
 
-@app.route('/convert', methods=['POST'])
-def convert():
-    file = request.files['file']
-    file_path = f"temp_{file.filename}"
-    file.save(file_path)
-    extract_and_convert(file_path)
-    return jsonify({"message": "Files converted successfully!"})
+# Streamlit UI
+st.title("File Converter Space")
+
+# Language selection
+language = st.selectbox("Select Language / انتخاب زبان", ["English", "فارسی"])
+
+if language == "English":
+    description = """
+    This space allows you to convert files between various formats.
+    Supported conversions:
+    - PDF to WEBP and back to PDF
+    - MP4 to MKV
+    - WAV to MP3
+    - PNG/JPG to WEBP
+    """
+    upload_label = "Choose a file"
+    convert_button = "Convert Files"
+    success_message = "Files converted successfully!"
+    converted_files_label = "Converted Files:"
+else:
+    description = """
+    این اسپیس به شما امکان می‌دهد فایل‌ها را بین فرمت‌های مختلف تبدیل کنید.
+    تبدیل‌های پشتیبانی شده:
+    - PDF به WEBP و بازگشت به PDF
+    - MP4 به MKV
+    - WAV به MP3
+    - PNG/JPG به WEBP
+    """
+    upload_label = "یک فایل انتخاب کنید"
+    convert_button = "تبدیل فایل‌ها"
+    success_message = "فایل‌ها با موفقیت تبدیل شدند!"
+    converted_files_label = "فایل‌های تبدیل شده:"
+
+st.write(description)
+
+uploaded_file = st.file_uploader(upload_label, type=['zip', 'rar'])
+
+if uploaded_file is not None:
+    file_path = f"temp_{uploaded_file.name}"
+    with open(file_path, "wb") as f:
+        f.write(uploaded_file.getbuffer())
+
+    st.success("File uploaded successfully!")
+
+    if st.button(convert_button):
+        extract_and_convert(file_path)
+        st.success(success_message)
 
-if __name__ == '__main__':
-    app.run(debug=True)
+        # Display converted files
+        st.write(converted_files_label)
+        extracted_dir = os.path.splitext(file_path)[0]
+        for root, _, files in os.walk(extracted_dir):
+            for file in files:
+                if file.endswith(('.pdf', '.mkv', '.mp3', '.webp')):
+                    st.write(f"- {file}")