Update api.py

api.py

@@ -1,126 +1,96 @@
-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
-
-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
-
-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 = []
-
-        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))
-
-        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
-
-    # در صورت نداشتن تغییر، همان مسیر بازگردانده می‌شود.
+#api.py
+
+import os import zipfile tempfile = import('tempfile') import shutil import rarfile from pdf2image import convert_from_path from PIL import Image from moviepy.editor import VideoFileClip from pydub import AudioSegment
+
+Allowed extensions for conversion
+
+ALLOWED_EXTENSIONS = {'zip', 'rar', 'pdf', 'wav', 'mp4', 'jpg', 'jpeg', 'png'}
+
+def allowed_file(filename): """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, output_dir=None): """ Convert a single file based on its extension. Returns the path to the converted file. """ base, ext = os.path.splitext(os.path.basename(input_path)) if output_dir is None: output_dir = os.path.dirname(input_path) ext = ext.lower()
+
+if ext == '.pdf':
+    images = convert_from_path(input_path)
+    out_pdf = os.path.join(output_dir, f"{base}_converted.pdf")
+    webp_imgs = []
+    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_imgs.append(Image.open(webp_path))
+    # Save all pages back to PDF
+    webp_imgs[0].save(out_pdf, 'PDF', save_all=True, append_images=webp_imgs[1:])
+    # Cleanup temporary WEBP files
+    for img in webp_imgs:
+        img.close()
+    for i in range(1, len(images)+1):
+        os.remove(os.path.join(output_dir, f"{base}_page_{i}.webp"))
+    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()
+    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:
+    # No conversion needed
     return input_path
 
-@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
-
-    try:
-        with tempfile.TemporaryDirectory() as temp_dir:
-            filename = secure_filename(file.filename)
-            upload_path = os.path.join(temp_dir, filename)
-            file.save(upload_path)
-
-            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
-
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+def extract_archive(input_path, extract_to): """Extract ZIP or RAR archives.""" if input_path.lower().endswith('.zip'): with zipfile.ZipFile(input_path, 'r') as z: z.extractall(extract_to) else: with rarfile.RarFile(input_path, 'r') as r: r.extractall(extract_to)
+
+def convert_path(input_path, work_dir): """ Convert a file or archive in work_dir. Returns a list of converted file paths. """ ext = os.path.splitext(input_path)[1].lower().lstrip('.') outputs = [] 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) # Convert each extracted file 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)) return outputs
+
+app.py
+
+import os import io import zipfile tempfile = import('tempfile') from werkzeug.utils import secure_filename import streamlit as st from api import ALLOWED_EXTENSIONS, allowed_file, convert_path
+
+Streamlit page configuration
+
+st.set_page_config(page_title="Universal File Converter", layout="wide")
+
+Multilingual interface text\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": "دانلود"
+}
+
+}
+
+Language selection
+
+lang = st.sidebar.selectbox("", list(TEXTS.keys())) T = TEXTS[lang]
+
+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 = convert_path(in_path, tmpdir) if len(converted) == 1: out_path = converted[0] with open(out_path, 'rb') as f: data = f.read() st.success(T['success']) st.download_button(T['download'], data, os.path.basename(out_path)) else: zip_buffer = io.BytesIO() with zipfile.ZipFile(zip_buffer, 'w') as zf: for p in converted: zf.write(p, arcname=os.path.basename(p)) st.success(T['success']) st.download_button(T['download'], zip_buffer.getvalue(), 'converted_files.zip')
+