Upload 3 files

api.py

@@ -0,0 +1,69 @@
+# 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__)
+
+def convert_file(input_path, output_path):
+    if input_path.endswith('.pdf'):
+        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')
+        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')
+    elif input_path.endswith('.wav'):
+        audio = AudioSegment.from_wav(input_path)
+        audio.export(output_path, format='mp3')
+    elif input_path.endswith(('.png', '.jpg')):
+        image = Image.open(input_path)
+        image.save(output_path, 'WEBP')
+
+def extract_and_convert(file_path):
+    if file_path.endswith('.zip'):
+        with zipfile.ZipFile(file_path, 'r') as zip_ref:
+            zip_ref.extractall(os.path.splitext(file_path)[0])
+            extracted_dir = os.path.splitext(file_path)[0]
+    elif file_path.endswith('.rar'):
+        with rarfile.RarFile(file_path, 'r') as rar_ref:
+            rar_ref.extractall(os.path.splitext(file_path)[0])
+            extracted_dir = os.path.splitext(file_path)[0]
+    else:
+        return
+
+    for root, _, files in os.walk(extracted_dir):
+        for file in files:
+            input_path = os.path.join(root, file)
+            output_path = os.path.join(root, os.path.splitext(file)[0] + get_output_extension(file))
+            convert_file(input_path, output_path)
+
+def get_output_extension(file):
+    if file.endswith('.pdf'):
+        return '.pdf'
+    elif file.endswith('.mp4'):
+        return '.mkv'
+    elif file.endswith('.wav'):
+        return '.mp3'
+    elif file.endswith(('.png', '.jpg')):
+        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!"})
+
+if __name__ == '__main__':
+    app.run(debug=True)

app.py

@@ -0,0 +1,121 @@
+# app.py
+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
+import streamlit as st
+
+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'):
+        # Convert MP4 to MKV
+        video = VideoFileClip(input_path)
+        video.write_videofile(output_path, codec='libx264')
+
+    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')
+
+def extract_and_convert(file_path):
+    if file_path.endswith('.zip'):
+        with zipfile.ZipFile(file_path, 'r') as zip_ref:
+            zip_ref.extractall(os.path.splitext(file_path)[0])
+            extracted_dir = os.path.splitext(file_path)[0]
+    elif file_path.endswith('.rar'):
+        with rarfile.RarFile(file_path, 'r') as rar_ref:
+            rar_ref.extractall(os.path.splitext(file_path)[0])
+            extracted_dir = os.path.splitext(file_path)[0]
+    else:
+        return
+
+    for root, _, files in os.walk(extracted_dir):
+        for file in files:
+            input_path = os.path.join(root, file)
+            output_path = os.path.join(root, os.path.splitext(file)[0] + get_output_extension(file))
+            convert_file(input_path, output_path)
+
+def get_output_extension(file):
+    if file.endswith('.pdf'):
+        return '.pdf'
+    elif file.endswith('.mp4'):
+        return '.mkv'
+    elif file.endswith('.wav'):
+        return '.mp3'
+    elif file.endswith(('.png', '.jpg')):
+        return '.webp'
+    return ''
+
+# 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)
+
+        # 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}")

space.yaml

@@ -0,0 +1,26 @@
+# space.yaml
+name: file-converter
+description: A space to convert files using various Python libraries.
+
+# Define the base image
+base: python:3.9-slim
+
+# Install system dependencies
+install:
+  - apt-get update
+  - apt-get install -y ffmpeg
+
+# Install Python dependencies
+pip:
+  - zipfile36
+  - rarfile
+  - PyPDF2
+  - moviepy
+  - pydub
+  - Pillow
+  - pdf2image
+  - streamlit
+  - flask
+
+# Define the main script to run
+entrypoint: streamlit run /path/to/your/app.py