Upload 3 files

api.py

@@ -1,78 +1,126 @@
-#api.py
-
-import os 
-import zipfile 
-import tempfile 
-import shutil 
-import rarfile 
-from pdf2image import convert_from_path 
-from PIL import Image 
-from moviepy.editor import VideoFileClip 
+from flask import Flask, request, send_file
+import os
+import zipfile
+import tempfile
+import shutil
+import rarfile
+from werkzeug.utils import secure_filename
+from pdf2image import convert_from_path
+from PIL import Image
+from moviepy.editor import VideoFileClip
 from pydub import AudioSegment
 
-Allowed extensions for conversion
+app = Flask(__name__)
+ALLOWED_EXTENSIONS = {'zip', 'rar', 'pdf', 'wav', 'mp4', 'jpg', 'png', 'jpeg'}
 
-ALLOWED_EXTENSIONS = {'zip', 'rar', 'pdf', 'wav', 'mp4', 'jpg', 'jpeg', 'png'}
+def allowed_file(filename):
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
-def allowed_file(filename: str) -> bool: """Check if the file extension is allowed.""" return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+def convert_single_file(input_path):
+    base_name, ext = os.path.splitext(input_path)
+    
+    if ext.lower() == '.pdf':
+        # PDF تبدیل به WEBP سپس به PDF
+        images = convert_from_path(input_path)
+        output_path = f"{base_name}_converted.pdf"
+        webp_images = []
 
-def convert_single_file(input_path: str, output_dir: str = None) -> str: """ Convert a single file based on its extension. Returns the path to the converted file. """ if output_dir is None: output_dir = os.path.dirname(input_path) base, ext = os.path.splitext(os.path.basename(input_path)) ext = ext.lower()
+        for i, image in enumerate(images):
+            webp_path = f"{base_name}_page_{i+1}.webp"
+            image.save(webp_path, 'WEBP')
+            webp_images.append(Image.open(webp_path))
 
-if ext == '.pdf':
-    images = convert_from_path(input_path)
-    out_pdf = os.path.join(output_dir, f"{base}_converted.pdf")
-    webp_images = []
-    for i, img in enumerate(images, start=1):
-        webp_path = os.path.join(output_dir, f"{base}_page_{i}.webp")
-        img.save(webp_path, 'WEBP')
-        webp_images.append(Image.open(webp_path))
-    webp_images[0].save(out_pdf, 'PDF', save_all=True, append_images=webp_images[1:])
-    # Cleanup
-    for img in webp_images:
-        img.close()
-    for i in range(1, len(images) + 1):
-        try:
-            os.remove(os.path.join(output_dir, f"{base}_page_{i}.webp"))
-        except OSError:
-            pass
-    return out_pdf
-
-elif ext == '.mp4':
-    out_vid = os.path.join(output_dir, f"{base}.mkv")
-    clip = VideoFileClip(input_path)
-    clip.write_videofile(out_vid, codec='libx264', audio_codec='aac')
-    clip.reader.close()
-    if clip.audio:
-        clip.audio.reader.close_proc()
-    return out_vid
-
-elif ext == '.wav':
-    out_audio = os.path.join(output_dir, f"{base}.mp3")
-    AudioSegment.from_wav(input_path).export(out_audio, format='mp3')
-    return out_audio
-
-elif ext in {'.png', '.jpg', '.jpeg'}:
-    out_img = os.path.join(output_dir, f"{base}.webp")
-    Image.open(input_path).save(out_img, 'WEBP')
-    return out_img
-
-else:
+        webp_images[0].save(output_path, 'PDF', save_all=True, append_images=webp_images[1:])
+        return output_path
+
+    elif ext.lower() == '.mp4':
+        # MP4 تبدیل به MKV
+        output_path = f"{base_name}.mkv"
+        VideoFileClip(input_path).write_videofile(output_path, codec='libx264')
+        return output_path
+
+    elif ext.lower() == '.wav':
+        # WAV تبدیل به MP3
+        output_path = f"{base_name}.mp3"
+        AudioSegment.from_wav(input_path).export(output_path, format='mp3')
+        return output_path
+
+    elif ext.lower() in {'.png', '.jpg', '.jpeg'}:
+        # تصویر تبدیل به WEBP
+        output_path = f"{base_name}.webp"
+        Image.open(input_path).save(output_path, 'WEBP')
+        return output_path
+
+    # در صورت نداشتن تغییر، همان مسیر بازگردانده می‌شود.
     return input_path
 
-def extract_archive(input_path: str, extract_to: str) -> None: """Extract ZIP or RAR archives.""" if input_path.lower().endswith('.zip'): with zipfile.ZipFile(input_path, 'r') as zf: zf.extractall(extract_to) else: with rarfile.RarFile(input_path, 'r') as rf: rf.extractall(extract_to)
+@app.route('/convert', methods=['POST'])
+def handle_conversion():
+    if 'file' not in request.files:
+        return {"error": "No file uploaded"}, 400
+
+    file = request.files['file']
+    if not file or file.filename == '':
+        return {"error": "Empty filename"}, 400
+
+    if not allowed_file(file.filename):
+        return {"error": "Unsupported file type"}, 400
 
-def convert_path(input_path: str, work_dir: str) -> list[str]: """ Convert a file or archive located at input_path within work_dir. Returns a list of paths to converted files. """ outputs: list[str] = [] ext = os.path.splitext(input_path)[1].lower().lstrip('.')
+    try:
+        with tempfile.TemporaryDirectory() as temp_dir:
+            filename = secure_filename(file.filename)
+            upload_path = os.path.join(temp_dir, filename)
+            file.save(upload_path)
 
-if ext in {'zip', 'rar'}:
-    extract_dir = os.path.join(work_dir, 'extracted')
-    os.makedirs(extract_dir, exist_ok=True)
-    extract_archive(input_path, extract_dir)
-    for root, _, files in os.walk(extract_dir):
-        for fname in files:
-            fpath = os.path.join(root, fname)
-            if allowed_file(fname):
-                outputs.append(convert_single_file(fpath, work_dir))
-else:
-    outputs.append(convert_single_file(input_path, work_dir))
+            file_ext = filename.split('.')[-1].lower()
+
+            # اگر فایل تکی باشد، عملیات تبدیل روی همان فایل اعمال شود.
+            if file_ext in {'pdf', 'wav', 'mp4', 'jpg', 'png', 'jpeg'}:
+                converted_path = convert_single_file(upload_path)
+                converted_filename = os.path.basename(converted_path)
+                return send_file(
+                    converted_path,
+                    as_attachment=True,
+                    download_name=converted_filename
+                )
+
+            # در صورتی که فایل آرشیو باشد: ابتدا استخراج شود
+            if filename.endswith('.zip'):
+                with zipfile.ZipFile(upload_path, 'r') as z:
+                    z.extractall(temp_dir)
+            else:
+                with rarfile.RarFile(upload_path, 'r') as r:
+                    r.extractall(temp_dir)
+
+            # تبدیل تمام فایل‌های موجود در آرشیو
+            for root, _, files in os.walk(temp_dir):
+                for f in files:
+                    if f != filename:
+                        input_file = os.path.join(root, f)
+                        convert_single_file(input_file)
+
+            # ایجاد فایل آرشیو خروجی
+            output_path = os.path.join(temp_dir, "converted.zip")
+            with zipfile.ZipFile(output_path, 'w') as z:
+                for root, _, files in os.walk(temp_dir):
+                    for f in files:
+                        if f != filename:
+                            full_path = os.path.join(root, f)
+                            z.write(full_path, arcname=f)
+
+            return send_file(
+                output_path,
+                as_attachment=True,
+                download_name="converted_files.zip"
+            )
+
+    except Exception as e:
+        return {"error": f"Processing failed: {str(e)}"}, 500
+    finally:
+        try:
+            shutil.rmtree(temp_dir, ignore_errors=True)
+        except Exception:
+            pass
 
-return outputs
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)

app.py

@@ -1,51 +1,97 @@
-#app.py
-
 import os
-import io
-import zipfile
-import tempfile
+import subprocess
+import requests
+from threading import Thread
+from flask import Flask, request, send_file
 from werkzeug.utils import secure_filename
+from pdf2image import convert_from_path
+from PIL import Image
+from moviepy.editor import VideoFileClip
+from pydub import AudioSegment
 import streamlit as st
-from api import ALLOWED_EXTENSIONS, allowed_file, convert_path
 
-Streamlit page configuration
+# Fix werkzeug compatibility for older versions if needed
+from werkzeug.urls import url_quote
+
+# Flask API Setup (برای اینجا فقط به هدف اجرا به عنوان پس‌زمینه استفاده می‌شود)
+app = Flask(__name__)
+ALLOWED_EXTENSIONS = {'zip', 'rar', 'pdf', 'wav', 'mp4', 'jpg', 'png', 'jpeg'}
+
+def allowed_file(filename):
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
-st.set_page_config(page_title="Universal File Converter", layout="wide")
+# راه‌اندازی API به عنوان پردازش پس‌زمینه
+def run_api():
+    # اجرای فایل api.py به صورت subprocess.
+    subprocess.run(['python', 'api.py'])
 
-Multilingual interface text
+# Streamlit UI
+def streamlit_app():
+    st.set_page_config(page_title="File Converter", layout="wide")
 
-TEXTS = { "English": { "title": "Universal File Converter", "desc": "Convert files or archives (ZIP, RAR, PDF, WAV, MP4, Images)", "upload": "Upload File", "convert_btn": "Convert", "processing": "Processing...", "error_type": "Unsupported file type!", "success": "Conversion successful!", "download": "Download" }, "فارسی": { "title": "مبدل جهانی فایل", "desc": "تبدیل فایل یا آرشیو (ZIP, RAR, PDF, WAV, MP4, تصاویر)", "upload": "آپلود فایل", "convert_btn": "تبدیل", "processing": "در حال پردازش...", "error_type": "نوع فایل پشتیبانی نمیشود!", "success": "تبدیل با موفقیت انجام شد!", "download": "دانلود" } }
+    TEXTS = {
+        "English": {
+            "title": "Universal File Converter",
+            "desc": "Convert files/archives (ZIP/RAR/PDF/WAV/MP4/IMG)",
+            "upload": "Upload File",
+            "error_type": "Unsupported file type!",
+            "success": "Conversion Successful!",
+            "download": "Download"
+        },
+        "فارسی": {
+            "title": "مبدل جهانی فایل",
+            "desc": "تبدیل فایل/آرشیو (ZIP/RAR/PDF/WAV/MP4/تصویر)",
+            "upload": "آپلود فایل",
+            "error_type": "نوع فایل پشتیبانی نمیشود!",
+            "success": "تبدیل با موفقیت انجام شد!",
+            "download": "دانلود"
+        }
+    }
 
-def main(): lang = st.sidebar.selectbox("", list(TEXTS.keys())) T = TEXTS[lang]
+    lang = st.sidebar.selectbox("", ["English", "فارسی"])
+    t = TEXTS[lang]
 
-st.title(T['title'])
-st.markdown(f"**{T['desc']}**")
+    st.title(t["title"])
+    st.markdown(f"**{t['desc']}**")
 
-uploaded = st.file_uploader(T['upload'], type=list(ALLOWED_EXTENSIONS))
-if uploaded:
-    fname = secure_filename(uploaded.name)
-    if not allowed_file(fname):
-        st.error(T['error_type'])
-    elif st.button(T['convert_btn']):
-        with st.spinner(T['processing']):
-            with tempfile.TemporaryDirectory() as tmpdir:
-                in_path = os.path.join(tmpdir, fname)
-                with open(in_path, 'wb') as f:
-                    f.write(uploaded.getbuffer())
-                converted_files = convert_path(in_path, tmpdir)
+    uploaded = st.file_uploader(t["upload"], type=ALLOWED_EXTENSIONS)
 
-                if len(converted_files) == 1:
-                    out_path = converted_files[0]
-                    with open(out_path, 'rb') as f2:
-                        data = f2.read()
-                    st.success(T['success'])
-                    st.download_button(T['download'], data, os.path.basename(out_path))
+    if uploaded:
+        file_extension = uploaded.name.split('.')[-1].lower()
+        if file_extension not in ALLOWED_EXTENSIONS:
+            st.error(t["error_type"])
+        elif st.button("Convert"):
+            try:
+                with st.spinner("Processing..."):
+                    response = requests.post(
+                        "http://localhost:7860/convert",
+                        files={"file": (uploaded.name, uploaded.getvalue(), uploaded.type)},
+                        timeout=300
+                    )
+
+                if response.status_code == 200:
+                    # استخراج نام فایل خروجی از header Content-Disposition
+                    cd = response.headers.get('Content-Disposition', '')
+                    filename = None
+                    if 'filename=' in cd:
+                        filename = cd.split("filename=")[1].strip().strip('"')
+                    else:
+                        filename = f"converted_{uploaded.name}"
+                        
+                    st.success(t["success"])
+                    st.download_button(
+                        label=t["download"],
+                        data=response.content,
+                        file_name=filename,
+                    )
                 else:
-                    zip_buffer = io.BytesIO()
-                    with zipfile.ZipFile(zip_buffer, 'w') as zf:
-                        for p in converted_files:
-                            zf.write(p, arcname=os.path.basename(p))
-                    st.success(T['success'])
-                    st.download_button(T['download'], zip_buffer.getvalue(), 'converted_files.zip')
-
-if name == 'main': main()
+                    st.error(f"Error {response.status_code}: {response.text}")
+
+            except Exception as e:
+                st.error(f"Connection error: {str(e)}")
+
+if __name__ == '__main__':
+    # اجرای API داخل یک ترد پس‌زمینه (در صورتی که نیاز به اجرای همزمان داشته باشید)
+    Thread(target=run_api, daemon=True).start()
+    # اجرای رابط کاربری Streamlit
+    streamlit_app()

requirements.txt

@@ -1,10 +1,11 @@
-Flask
-python-magic
-requests
-streamlit
-pdf2image
-Pillow
-moviepy
-pydub
-rarfile
-werkzeug
+# requirements.txt
+Flask==2.0.3
+Werkzeug==2.0.3  # نسخه سازگار با url_quote
+streamlit==1.22.0
+python-magic==0.4.27
+pdf2image==1.16.3
+Pillow==10.0.1
+pydub==0.25.1
+moviepy==1.0.3
+rarfile==4.0
+requests