Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, request, jsonify
+from flask import Flask, render_template, request, jsonify, Response
 from flask_cors import CORS
 import internetarchive as ia
 import os
@@ -7,7 +7,6 @@ import uuid
 import threading
 import time
 import requests
-import json
 
 app = Flask(__name__)
 CORS(app, resources={r"/*": {"origins": "*"}})
@@ -19,6 +18,7 @@ def after_request(response):
     response.headers.add('Access-Control-Allow-Methods', 'GET,POST,OPTIONS')
     return response
 
+# Background pinger to keep file2link warm
 def ping_file2link():
     while True:
         try:
@@ -28,6 +28,7 @@ def ping_file2link():
             print(f"[{datetime.now()}] Ping error: {str(e)}")
         time.sleep(10)
 
+# Start the pinger thread
 threading.Thread(target=ping_file2link, daemon=True).start()
 
 @app.route('/')
@@ -57,7 +58,7 @@ def upload_file():
             r = ia.upload(
                 item_id,
                 files={unique_filename: file},
-                metadata=None,  # Skip metadata
+                metadata=None,
                 access_key='YXzY5OrREXLL6XBh',
                 secret_key='m2XnL3X7xCB1pNGE',
                 retries=5,
@@ -65,10 +66,11 @@ def upload_file():
             )
 
             if r[0].status_code == 200:
-                access_url = f'https://archive.org/download/{item_id}/{unique_filename}'
+                # Instead of Archive.org link, return masked Hugging Face Space URL
+                hf_url = f'https://shashwatidrnpo-inv.hf.space/files/serve/{unique_filename}'
                 return jsonify({
                     'success': True,
-                    'access_url': access_url,
+                    'access_url': hf_url,
                     'item_id': item_id,
                     'original_filename': original_filename,
                     'unique_filename': unique_filename
@@ -79,6 +81,24 @@ def upload_file():
         except Exception as e:
             return jsonify({'error': str(e)}), 500
 
+@app.route('/files/serve/<filename>')
+def serve_file(filename):
+    archive_url = f'https://archive.org/download/VELTRIX_DB/{filename}'
+    try:
+        r = requests.get(archive_url, stream=True)
+        return Response(
+            r.iter_content(chunk_size=8192),
+            headers={
+                'Content-Type': r.headers.get('Content-Type', 'application/octet-stream'),
+                'Content-Disposition': f'inline; filename="{filename}"',
+                'Accept-Ranges': 'bytes',
+                'Cache-Control': 'public, max-age=604800'
+            },
+            status=r.status_code
+        )
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
 if __name__ == '__main__':
     port = int(os.environ.get('PORT', 5000))
     app.run(host='0.0.0.0', port=port, debug=True)