Update app.py

app.py

@@ -1,97 +1,67 @@
+# app.py
 import os
-import subprocess
-import requests
-from threading import Thread
-from flask import Flask, request, send_file
+import io
+import zipfile
+tempfile = __import__('tempfile')
 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
 
-# Fix werkzeug compatibility for older versions if needed
-from werkzeug.urls import url_quote
+# Streamlit page configuration
+st.set_page_config(page_title="Universal File Converter", layout="wide")
 
-# 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
-
-# راه‌اندازی API به عنوان پردازش پس‌زمینه
-def run_api():
-    # اجرای فایل api.py به صورت subprocess.
-    subprocess.run(['python', 'api.py'])
-
-# Streamlit UI
-def streamlit_app():
-    st.set_page_config(page_title="File Converter", layout="wide")
-
-    TEXTS = {
-        "English": {
-            "title": "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": "دانلود"
-        }
+# 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": "دانلود"
     }
+}
 
-    lang = st.sidebar.selectbox("", ["English", "فارسی"])
-    t = TEXTS[lang]
+# Language selection
+lang = st.sidebar.selectbox("", list(TEXTS.keys()))
+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=ALLOWED_EXTENSIONS)
-
-    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,
-                    )
+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:
-                    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()
+                    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')