Update app.py

app.py

@@ -29,7 +29,7 @@ if not HF_TOKEN:
 hf_api = HfApi(token=HF_TOKEN)
 logger.info(f"Public File Upload API Initialized. Target repo: {HF_REPO_ID}")
 
-# --- توابع کمکی (بدون تغییر) ---
+# --- توابع کمکی ---
 def get_folder_by_mimetype(mimetype):
     if not mimetype: return "others"
     main_type = mimetype.split('/')[0]
@@ -64,7 +64,7 @@ def serve_html():
 @app.route('/upload', methods=['POST'])
 def upload_file_endpoint():
     try:
-        # آپلود از کامپیوتر یا ربات (حالت FormData)
+        # حالت اول: آپلود از کامپیوتر یا ربات (حالت FormData)
         if 'file' in request.files:
             uploaded_file = request.files['file']
             if uploaded_file.filename == '':
@@ -72,8 +72,32 @@ def upload_file_endpoint():
             file_content = uploaded_file.read()
             direct_url = process_and_upload(file_content, uploaded_file.filename, uploaded_file.mimetype)
             return jsonify({"message": "File uploaded successfully!", "hf_url": direct_url}), 200
+
+        # ***** این بخش دوباره اضافه شد *****
+        # حالت دوم: آپلود از طریق لینک در وب سایت (حالت JSON)
+        elif request.is_json and 'url' in request.json:
+            file_url = request.json['url']
+            if not file_url:
+                return jsonify({"error": "No URL provided"}), 400
+
+            logger.info(f"Downloading file from URL: {file_url}")
+            response = requests.get(file_url, stream=True, timeout=30)
+            response.raise_for_status()
+            
+            file_content = response.content
+            mimetype = response.headers.get('Content-Type')
+            original_filename = file_url.split('/')[-1].split('?')[0]
+
+            direct_url = process_and_upload(file_content, original_filename, mimetype)
+            return jsonify({"message": "File uploaded successfully!", "hf_url": direct_url}), 200
+        # ***** پایان بخش اضافه شده *****
+            
         else:
-            return jsonify({"error": "Invalid request. 'file' part is missing."}), 400
+            return jsonify({"error": "Invalid request. Expected 'file' (form-data) or 'url' (json)."}), 400
+
+    except requests.exceptions.RequestException as e:
+        logger.error(f"Failed to download from URL: {e}", exc_info=True)
+        return jsonify({"error": f"Failed to download from URL: {str(e)}"}), 400
     except Exception as e:
         logger.error(f"An internal server error occurred: {e}", exc_info=True)
         return jsonify({"error": f"An internal server error occurred: {str(e)}"}), 500