Update app.py

app.py

@@ -1,21 +1,92 @@
-import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
-from huggingface_hub import login
-import torch
+from flask import Flask, render_template, request, jsonify, Response
+from flask_cors import CORS
+import internetarchive as ia
 import os
+from datetime import datetime
+import uuid
+import threading
+import time
+import requests
+import json
 
-# Use token from environment variable (set securely in Hugging Face Space)
-token = os.environ.get("HF_TOKEN")
-login(token)
+app = Flask(__name__)
+CORS(app)
 
-model_id = "bigcode/starcoder"
-tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=token)
-model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto", use_auth_token=token)
+@app.after_request
+def after_request(response):
+    response.headers.add('Access-Control-Allow-Origin', '*')
+    response.headers.add('Access-Control-Allow-Headers', 'Content-Type,Authorization')
+    response.headers.add('Access-Control-Allow-Methods', 'GET,PUT,POST,DELETE,OPTIONS')
+    return response
 
-def generate_code(prompt):
-    inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
-    outputs = model.generate(**inputs, max_new_tokens=200)
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+def ping_file2link():
+    while True:
+        try:
+            response = requests.get('https://file2link-3pse.onrender.com')
+            print(f"[{datetime.now()}] Ping status: {response.status_code}")
+        except Exception as e:
+            print(f"[{datetime.now()}] Ping error: {str(e)}")
+        time.sleep(10)
 
-iface = gr.Interface(fn=generate_code, inputs="text", outputs="text", title="StarCoder Code Generator")
-iface.launch(server_name="0.0.0.0", server_port=7860)
+threading.Thread(target=ping_file2link, daemon=True).start()
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+@app.route('/upload', methods=['POST', 'OPTIONS'])
+def upload_file():
+    if request.method == 'OPTIONS':
+        return '', 200
+
+    if 'file' not in request.files:
+        return jsonify({'error': 'No file part'}), 400
+
+    file = request.files['file']
+    if file.filename == '':
+        return jsonify({'error': 'No selected file'}), 400
+
+    if file:
+        item_id = 'shop_proj'
+        original_filename = file.filename
+        file_extension = os.path.splitext(original_filename)[1]
+        unique_filename = f"{uuid.uuid4()}{file_extension}"
+
+        metadata = {
+            'title': f'File2Link Upload - {original_filename}',
+            'mediatype': 'data',
+            'collection': 'opensource',
+            'description': f'Original filename: {original_filename}',
+            'creator': 'File2Link Uploader',
+            'date': datetime.now().strftime('%Y-%m-%d'),
+            'licenseurl': 'https://creativecommons.org/licenses/by/4.0/',
+            'subject': ['file2link', 'file sharing'],
+            'original_filename': original_filename
+        }
+
+        def generate_progress():
+            try:
+                file.seek(0)
+                r = ia.upload(
+                    item_id,
+                    files={unique_filename: file},
+                    metadata=metadata,
+                    access_key='PrJnoIKjNt4ul1Fr',         # <-- HARDCODED
+                    secret_key='S0tCXWb7fM43m44Y'          # <-- HARDCODED
+                )
+                if r[0].status_code == 200:
+                    access_url = f'https://archive.org/download/{item_id}/{unique_filename}'
+                    return json.dumps({
+                        'success': True,
+                        'access_url': access_url,
+                        'item_id': item_id,
+                        'original_filename': original_filename,
+                        'unique_filename': unique_filename,
+                        'metadata': metadata
+                    })
+                else:
+                    return json.dumps({'error': 'Upload failed'})
+            except Exception as e:
+                return json.dumps({'error': str(e)})
+
+        return Response(generate_progress(), mimetype='application/json')