Update app.py

app.py

@@ -3,7 +3,7 @@ import uuid
 import requests
 import threading
 import time
-from flask import Flask, request, jsonify, render_template, Response
+from flask import Flask, request, jsonify, render_template
 from werkzeug.utils import secure_filename
 from itertools import cycle
 from dotenv import load_dotenv
@@ -41,15 +41,12 @@ def get_permanent_link(job_id, temp_render_url, worker_url):
         response.raise_for_status()
         data = response.json()
         
-        uploader_proxy_url = data.get("hf_url")
-        if not uploader_proxy_url: raise Exception("Invalid response from uploader")
-        
-        path_part = uploader_proxy_url.split('/proxy/')[1]
-        final_proxy_url = f"/proxy/{path_part}"
+        final_url = data.get("hf_url") or data.get("url")
+        if not final_url: raise Exception("Invalid response from uploader service")
         
         with lock:
             jobs[job_id]["status"] = "completed"
-            jobs[job_id]["result"] = final_proxy_url
+            jobs[job_id]["result"] = final_url
             
     except Exception as e:
         print(f"Error making link permanent for {job_id}: {e}")
@@ -152,23 +149,5 @@ def check_status():
     with lock: job = jobs.get(job_id, {"status": "not_found", "result": None})
     return jsonify({"status": job["status"], "result": job["result"]})
 
-@app.route('/proxy/<user>/<repo>/<path:file_path>')
-def file_proxy(user, repo, file_path):
-    if not HF_TOKEN: return "HF_TOKEN not set", 500
-    repo_id = f"{user}/{repo}"
-    file_url = f"https://huggingface.co/datasets/{repo_id}/resolve/main/{file_path}"
-    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-    try:
-        hf_response = requests.get(file_url, headers=headers, stream=True, timeout=30)
-        hf_response.raise_for_status()
-        return Response(hf_response.iter_content(chunk_size=8192),
-                        content_type=hf_response.headers.get('Content-Type', 'application/octet-stream'))
-    except requests.exceptions.HTTPError as e:
-        if e.response.status_code == 404: return "File not found", 404
-        return f"Hugging Face Server Error: {e.response.status_code}", e.response.status_code
-    except Exception as e:
-        print(f"Proxy error: {e}")
-        return "Error fetching file", 500
-
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))