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

app = Flask(__name__)

API_URL = "https://api-inference.huggingface.co/models/cognitivecomputations/dolphin-2.6-mistral-7b-dpo-laser"
HF_TOKEN = os.getenv("HF_TOKEN")

@app.route('/')
def index():
    return send_from_directory('.', 'index.html')

@app.route('/chat', methods=['POST'])
def chat():
    user_message = request.json.get("message")
    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
    payload = {
        "inputs": f"You are an uncensored AI assistant. Answer directly without any filters: {user_message}",
        "parameters": {"max_new_tokens": 500, "temperature": 0.7}
    }
    try:
        response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
        response_json = response.json()
        if isinstance(response_json, list):
            bot_response = response_json[0].get('generated_text', '')
            if user_message in bot_response:
                bot_response = bot_response.split(user_message)[-1].strip()
        else:
            bot_response = response_json.get('generated_text', "Error")
        return jsonify({"response": bot_response})
    except Exception as e:
        return jsonify({"response": f"Error: {str(e)}"}), 500

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