from flask import Flask, send_file, session, render_template_string
from PIL import Image, ImageDraw, ImageFont
import random
import string
import io

app = Flask(name)
app.secret_key = 'your_secret_key'  # کلید مخفی برای نشست

# تابع برای تولید کپچا
def generate_captcha():
    # تولید رشته تصادفی
    letters = string.ascii_uppercase + string.digits
    captcha_text = ''.join(random.choices(letters, k=6))  # 6 کاراکتر تصادفی

    # ایجاد تصویر
    image = Image.new('RGB', (200, 70), (255, 255, 255))
    draw = ImageDraw.Draw(image)

    # استفاده از فونت پیش‌فرض
    font = ImageFont.load_default()
   
    # ترسیم متن کپچا
    draw.text((30, 15), captcha_text, fill=(0, 0, 0), font=font)

    # ذخیره کپچا در حافظه
    buf = io.BytesIO()
    image.save(buf, format='PNG')
    buf.seek(0)

    session['captcha'] = captcha_text  # ذخیره کپچا در جلسه
    return buf

# صفحه اصلی
@app.route('/')
def index():
    return render_template_string('''
        <h1>کپچا تولید شده:</h1>
        <img src="/captcha" alt="CAPTCHA">
        <form method="post" action="/submit">
            <input type="text" name="captcha_input" placeholder="کپچا را وارد کنید" required>
            <button type="submit">ارسال</button>
        </form>
    ''')

# تولید تصویر کپچا
@app.route('/captcha')
def captcha():
    img = generate_captcha()
    return send_file(img, mimetype='image/png')

# پردازش ورودی کاربر
@app.route('/submit', methods=['POST'])
def submit():
    captcha_input = request.form['captcha_input']
    if captcha_input == session.get('captcha'):
        return "کپچا صحیح است!"
    else:
        return "کپچا نادرست است. لطفاً دوباره امتحان کنید."

if name == 'main':
    app.run(debug=True)