from flask import Flask, request, jsonify
import requests
import os

app = Flask(__name__)

OPENROUTER_KEY = os.getenv("OPENROUTER_KEY", "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")

@app.route("/api/send", methods=["POST"])
def send():
    data = request.get_json()
    prompt = data.get("prompt")
    model = data.get("model", "deepseek/deepseek-r1:free")

    try:
        response = requests.post(
            "https://openrouter.ai/api/v1/chat/completions",
            headers={
                "Authorization": f"Bearer {OPENROUTER_KEY}",
                "Content-Type": "application/json",
                "HTTP-Referer": "https://huggingface.co/spaces/studycode129/Free_Web_LLM_Tester"
            },
            json={
                "model": model,
                "messages": [{"role": "user", "content": prompt}],
                "temperature": 0.7
            }
        )
        return jsonify(response.json())
    except Exception as e:
        return jsonify({"error": str(e)}), 500


if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860)