import os
os.environ["QT_QPA_PLATFORM"] = "offscreen"
import cv2
import numpy as np
import pytesseract
from PIL import Image
from flask import Flask, request, send_file, jsonify
from rembg import remove
import io
import asyncio
import edge_tts

app = Flask(__name__)

@app.route('/')
def home():
    return "🔥 Daimond Batch Heavy Engine is RUNNING! 🔥"

# 1. Background Remover API
@app.route('/bg-remove', methods=['POST'])
def api_bg_remove():
    try:
        file = request.files['image'].read()
        output = remove(file)
        return send_file(io.BytesIO(output), mimetype='image/png')
    except Exception as e:
        return jsonify({"error": str(e)}), 500

# 2. Pencil Sketch API
@app.route('/sketch', methods=['POST'])
def api_sketch():
    try:
        file = request.files['image'].read()
        npimg = np.frombuffer(file, np.uint8)
        img = cv2.imdecode(npimg, cv2.IMREAD_COLOR)
        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        invert = cv2.bitwise_not(gray)
        blur = cv2.GaussianBlur(invert, (21, 21), 0)
        sketch = cv2.divide(gray, 255 - blur, scale=256.0)
        is_success, buffer = cv2.imencode(".jpg", sketch)
        return send_file(io.BytesIO(buffer), mimetype='image/jpeg')
    except Exception as e:
        return jsonify({"error": str(e)}), 500

# 3. Optical Text Reader (OCR) API
@app.route('/ocr', methods=['POST'])
def api_ocr():
    try:
        file = request.files['image']
        img = Image.open(file)
        text = pytesseract.image_to_string(img)
        return jsonify({"text": text.strip()})
    except Exception as e:
        return jsonify({"error": str(e)}), 500

# 4. 🎙️ VIP Voice Generator API (Direct Python)
@app.route('/tts', methods=['POST'])
def tts_api():
    try:
        text = request.form.get('text', 'Hello')
        rate = request.form.get('rate', '+0%')
        # 👇 Yahan 'SwaraNeural' laga diya hai jo ekdum sweet Indian female voice hai
        voice = request.form.get('voice', 'hi-IN-SwaraNeural') 

        audio_file = "temp_tts.ogg"

        async def generate_audio():
            communicate = edge_tts.Communicate(text, voice, rate=rate)
            await communicate.save(audio_file)
            
        asyncio.run(generate_audio())
        return send_file(audio_file, mimetype="audio/ogg")
    except Exception as e:
        return str(e), 500

# HF ki app.py mein ye naya endpoint jodo
@app.route('/enhance', methods=['POST'])
def enhance_image():
    try:
        file = request.files['image']
        img = Image.open(file.stream)
        
        # Yahan hum GFPGAN ya RealESRGAN model ko call karenge
        # Filhal main ek high-quality sharpening logic de raha hoon 
        # Jo bina kisi heavy model ke bhi photo ko clean kar dega
        from PIL import ImageEnhance, ImageFilter
        
        # Detail badhane ke liye
        img = img.filter(ImageFilter.DETAIL)
        
        # Sharpness badhane ke liye
        enhancer = ImageEnhance.Sharpness(img)
        img = enhancer.enhance(2.0)
        
        # Color thode vibrant karne ke liye
        color_enhancer = ImageEnhance.Color(img)
        img = color_enhancer.enhance(1.2)
        
        byte_io = io.BytesIO()
        img.save(byte_io, 'JPEG', quality=100)
        byte_io.seek(0)
        
        return send_file(byte_io, mimetype='image/jpeg')
    except Exception as e:
        return str(e), 500        

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