Update app.py

app.py

@@ -1,85 +1,891 @@
+import os
+import re
+import sqlite3
+import hashlib
+import secrets
+from datetime import date
+
 import gradio as gr
 import torch
+
 from transformers import AutoTokenizer
-from peft import AutoPeftModelForSequenceClassification
+from peft import AutoPeftModelForCausalLM
 
-# =========================
-# MODEL
-# =========================
+
+# ============================================================
+# CONFIG
+# ============================================================
 
 MODEL_PATH = "."
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
+DB_PATH = "users.db"
+
+FREE_DAILY_LIMIT = 10
+PRO_DAILY_LIMIT = 1000
+
+ADMIN_EMAIL = os.getenv("ADMIN_EMAIL", "admin@example.com")
+ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD", "CHANGE_THIS_PASSWORD")
+
+
+# ============================================================
+# DATABASE
+# ============================================================
+
+def db():
+    conn = sqlite3.connect(DB_PATH, check_same_thread=False)
+    conn.row_factory = sqlite3.Row
+    return conn
+
+
+def init_db():
+
+    conn = db()
+
+    conn.execute("""
+        CREATE TABLE IF NOT EXISTS users (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            email TEXT UNIQUE NOT NULL,
+            password_hash TEXT NOT NULL,
+            salt TEXT NOT NULL,
+            plan TEXT DEFAULT 'free',
+            active INTEGER DEFAULT 1,
+            daily_usage INTEGER DEFAULT 0,
+            usage_date TEXT,
+            created_at TEXT DEFAULT CURRENT_TIMESTAMP
+        )
+    """)
+
+    conn.commit()
+
+    # Admin hesabını otomatik oluştur
+    existing = conn.execute(
+        "SELECT id FROM users WHERE email = ?",
+        (ADMIN_EMAIL,)
+    ).fetchone()
+
+    if not existing:
+        password_hash, salt = hash_password(ADMIN_PASSWORD)
+
+        conn.execute("""
+            INSERT INTO users
+            (email, password_hash, salt, plan, active)
+            VALUES (?, ?, ?, 'admin', 1)
+        """, (
+            ADMIN_EMAIL,
+            password_hash,
+            salt
+        ))
+
+        conn.commit()
+
+    conn.close()
+
+
+# ============================================================
+# PASSWORD SECURITY
+# ============================================================
+
+def hash_password(password, salt=None):
+
+    if salt is None:
+        salt = secrets.token_hex(16)
+
+    password_hash = hashlib.pbkdf2_hmac(
+        "sha256",
+        password.encode("utf-8"),
+        salt.encode("utf-8"),
+        200_000
+    ).hex()
+
+    return password_hash, salt
+
+
+def verify_password(password, stored_hash, salt):
+
+    calculated, _ = hash_password(password, salt)
+
+    return secrets.compare_digest(
+        calculated,
+        stored_hash
+    )
+
+
+# ============================================================
+# USER FUNCTIONS
+# ============================================================
+
+def get_user(email):
+
+    conn = db()
+
+    user = conn.execute(
+        "SELECT * FROM users WHERE email = ?",
+        (email.lower().strip(),)
+    ).fetchone()
+
+    conn.close()
+
+    return user
+
+
+def create_user(email, password):
+
+    email = email.lower().strip()
+
+    if not re.match(
+        r"^[^@\s]+@[^@\s]+\.[^@\s]+$",
+        email
+    ):
+        return False, "Geçerli bir e-posta adresi gir."
+
+    if len(password) < 8:
+        return False, "Şifre en az 8 karakter olmalı."
+
+    if get_user(email):
+        return False, "Bu e-posta zaten kayıtlı."
+
+    password_hash, salt = hash_password(password)
+
+    conn = db()
+
+    conn.execute("""
+        INSERT INTO users
+        (email, password_hash, salt, plan, active)
+        VALUES (?, ?, ?, 'free', 1)
+    """, (
+        email,
+        password_hash,
+        salt
+    ))
+
+    conn.commit()
+    conn.close()
+
+    return True, "Hesap oluşturuldu. Giriş yapabilirsin."
+
+
+def login_user(email, password):
+
+    user = get_user(email)
+
+    if not user:
+        return None, "E-posta veya şifre hatalı."
+
+    if not verify_password(
+        password,
+        user["password_hash"],
+        user["salt"]
+    ):
+        return None, "E-posta veya şifre hatalı."
+
+    if not user["active"]:
+        return None, "Hesabın pasif durumda."
+
+    # Günlük kullanım reset
+    today = str(date.today())
+
+    if user["usage_date"] != today:
+
+        conn = db()
+
+        conn.execute("""
+            UPDATE users
+            SET daily_usage = 0,
+                usage_date = ?
+            WHERE id = ?
+        """, (
+            today,
+            user["id"]
+        ))
+
+        conn.commit()
+        conn.close()
+
+        user = get_user(email)
+
+    return user, "Giriş başarılı."
 
-tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
 
-model = AutoPeftModelForSequenceClassification.from_pretrained(
+# ============================================================
+# USAGE LIMIT
+# ============================================================
+
+def get_limit(plan):
+
+    if plan == "admin":
+        return 999999
+
+    if plan == "pro":
+        return PRO_DAILY_LIMIT
+
+    return FREE_DAILY_LIMIT
+
+
+def can_use(user):
+
+    today = str(date.today())
+
+    if user["usage_date"] != today:
+        return True
+
+    limit = get_limit(user["plan"])
+
+    return user["daily_usage"] < limit
+
+
+def increment_usage(user_id):
+
+    conn = db()
+
+    conn.execute("""
+        UPDATE users
+        SET daily_usage = daily_usage + 1,
+            usage_date = ?
+        WHERE id = ?
+    """, (
+        str(date.today()),
+        user_id
+    ))
+
+    conn.commit()
+    conn.close()
+
+
+# ============================================================
+# MODEL
+# ============================================================
+
+print("Loading model...")
+
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+
+print("Device:", DEVICE)
+
+tokenizer = AutoTokenizer.from_pretrained(
+    MODEL_PATH
+)
+
+model = AutoPeftModelForCausalLM.from_pretrained(
     MODEL_PATH,
-    torch_dtype=torch.float16 if device == "cuda" else torch.float32
+    torch_dtype=(
+        torch.float16
+        if DEVICE == "cuda"
+        else torch.float32
+    )
 )
 
-model.to(device)
+model.to(DEVICE)
 model.eval()
 
+print("Model loaded.")
+
+
+# ============================================================
+# SENTIMENT
+# ============================================================
+
+def extract_sentiment(text):
+
+    text_lower = text.lower()
+
+    if "bullish" in text_lower:
+        return "Bullish"
 
-# =========================
-# PREDICTION
-# =========================
+    if "bearish" in text_lower:
+        return "Bearish"
 
-def predict(text):
+    if "neutral" in text_lower:
+        return "Neutral"
+
+    return "Unknown"
+
+
+def analyze_crypto(text, user):
 
     if not text or not text.strip():
-        return {
-            "Bullish": 0.0,
-            "Neutral": 0.0,
-            "Bearish": 0.0
-        }
+        return "Metin gir."
+
+    # Kullanıcıyı tekrar DB'den kontrol et
+    current_user = get_user(user["email"])
+
+    if not current_user:
+        return "Oturum bulunamadı."
+
+    if not current_user["active"]:
+        return "Hesabın aktif değil."
+
+    if not can_use(current_user):
+
+        limit = get_limit(current_user["plan"])
+
+        return (
+            f"Günlük kullanım limitine ulaştın.\n\n"
+            f"Plan: {current_user['plan'].upper()}\n"
+            f"Limit: {limit}"
+        )
+
+    prompt = f"""
+You are a cryptocurrency financial sentiment classifier.
+
+Classify the following cryptocurrency-related text into exactly one
+of these labels:
+
+Bullish
+Neutral
+Bearish
+
+Return only one label.
+
+Text:
+{text}
+
+Sentiment:
+"""
 
     inputs = tokenizer(
-        text,
+        prompt,
         return_tensors="pt",
         truncation=True,
         max_length=512
     )
 
-    inputs = {k: v.to(device) for k, v in inputs.items()}
+    inputs = {
+        k: v.to(DEVICE)
+        for k, v in inputs.items()
+    }
 
     with torch.no_grad():
-        outputs = model(**inputs)
-        probabilities = torch.softmax(outputs.logits, dim=-1)[0]
 
-    labels = ["Bearish", "Neutral", "Bullish"]
+        output = model.generate(
+            **inputs,
+            max_new_tokens=10,
+            do_sample=False
+        )
 
-    result = {
-        labels[i]: float(probabilities[i])
-        for i in range(len(labels))
-    }
+    generated = output[
+        0
+    ][
+        inputs["input_ids"].shape[1]:
+    ]
 
-    return result
-
-
-# =========================
-# GRADIO UI
-# =========================
-
-demo = gr.Interface(
-    fn=predict,
-    inputs=gr.Textbox(
-        label="Crypto Text",
-        placeholder="Bitcoin ETF inflows increased sharply today...",
-        lines=5
-    ),
-    outputs=gr.Label(
-        label="Crypto Sentiment",
-        num_top_classes=3
-    ),
-    title="CryptoBERT Sentiment",
-    description="Bullish / Neutral / Bearish crypto sentiment analysis"
-)
+    result = tokenizer.decode(
+        generated,
+        skip_special_tokens=True
+    ).strip()
+
+    sentiment = extract_sentiment(result)
+
+    increment_usage(current_user["id"])
+
+    if sentiment == "Unknown":
+        return (
+            "Model sonucu:\n\n"
+            + result
+        )
+
+    return sentiment
+
+
+# ============================================================
+# AUTH UI
+# ============================================================
+
+def signup(email, password):
+
+    success, message = create_user(
+        email,
+        password
+    )
+
+    return message
+
+
+def login(email, password):
+
+    user, message = login_user(
+        email,
+        password
+    )
+
+    if not user:
+        return (
+            message,
+            None,
+            gr.update(visible=True),
+            gr.update(visible=False),
+            gr.update(visible=False),
+            ""
+        )
+
+    admin_visible = (
+        user["plan"] == "admin"
+    )
+
+    status = (
+        f"Hoş geldin: {user['email']}\n"
+        f"Plan: {user['plan'].upper()}\n"
+        f"Kullanım: {user['daily_usage']}/"
+        f"{get_limit(user['plan'])}"
+    )
+
+    return (
+        status,
+        dict(user),
+        gr.update(visible=False),
+        gr.update(visible=True),
+        gr.update(visible=admin_visible),
+        ""
+    )
+
+
+def logout():
+
+    return (
+        "",
+        None,
+        gr.update(visible=True),
+        gr.update(visible=False),
+        gr.update(visible=False),
+        ""
+    )
+
+
+# ============================================================
+# ADMIN
+# ============================================================
+
+def get_users_dataframe():
+
+    conn = db()
+
+    rows = conn.execute("""
+        SELECT
+            id,
+            email,
+            plan,
+            active,
+            daily_usage,
+            created_at
+        FROM users
+        ORDER BY id DESC
+    """).fetchall()
+
+    conn.close()
+
+    data = []
+
+    for row in rows:
+
+        data.append([
+            row["id"],
+            row["email"],
+            row["plan"],
+            "Active" if row["active"] else "Inactive",
+            row["daily_usage"],
+            row["created_at"]
+        ])
+
+    return data
+
+
+def refresh_users():
+
+    return get_users_dataframe()
+
+
+def admin_change_plan(user_id, plan):
+
+    try:
+        user_id = int(user_id)
+    except:
+        return "Geçersiz kullanıcı ID."
+
+    if plan not in [
+        "free",
+        "pro",
+        "admin"
+    ]:
+        return "Geçersiz plan."
+
+    conn = db()
+
+    conn.execute("""
+        UPDATE users
+        SET plan = ?
+        WHERE id = ?
+    """, (
+        plan,
+        user_id
+    ))
+
+    conn.commit()
+    conn.close()
+
+    return f"Kullanıcı {user_id} → {plan}"
+
+
+def admin_toggle_user(user_id):
+
+    try:
+        user_id = int(user_id)
+    except:
+        return "Geçersiz kullanıcı ID."
+
+    conn = db()
+
+    user = conn.execute(
+        "SELECT active FROM users WHERE id = ?",
+        (user_id,)
+    ).fetchone()
+
+    if not user:
+        conn.close()
+        return "Kullanıcı bulunamadı."
+
+    new_status = 0 if user["active"] else 1
+
+    conn.execute("""
+        UPDATE users
+        SET active = ?
+        WHERE id = ?
+    """, (
+        new_status,
+        user_id
+    ))
+
+    conn.commit()
+    conn.close()
+
+    return (
+        f"Kullanıcı {user_id}: "
+        + (
+            "Aktif"
+            if new_status
+            else "Pasif"
+        )
+    )
+
+
+def admin_delete_user(user_id):
+
+    try:
+        user_id = int(user_id)
+    except:
+        return "Geçersiz kullanıcı ID."
+
+    conn = db()
+
+    user = conn.execute(
+        "SELECT email FROM users WHERE id = ?",
+        (user_id,)
+    ).fetchone()
+
+    if not user:
+        conn.close()
+        return "Kullanıcı bulunamadı."
+
+    if user["email"] == ADMIN_EMAIL:
+        conn.close()
+        return "Ana admin silinemez."
+
+    conn.execute(
+        "DELETE FROM users WHERE id = ?",
+        (user_id,)
+    )
+
+    conn.commit()
+    conn.close()
+
+    return f"Kullanıcı {user_id} silindi."
+
+
+# ============================================================
+# INITIALIZE
+# ============================================================
+
+init_db()
+
+
+# ============================================================
+# GRADIO
+# ============================================================
+
+with gr.Blocks(
+    title="CryptoBERT AI"
+) as demo:
+
+    session = gr.State(None)
+
+    # ========================================================
+    # LOGIN
+    # ========================================================
+
+    with gr.Column(
+        visible=True
+    ) as login_panel:
+
+        gr.Markdown(
+            """
+# 🚀 CryptoBERT AI
+
+Kripto piyasası sentiment analiz sistemi.
+
+**Bullish / Neutral / Bearish**
+"""
+        )
+
+        email = gr.Textbox(
+            label="E-posta",
+            placeholder="email@example.com"
+        )
+
+        password = gr.Textbox(
+            label="Şifre",
+            type="password"
+        )
+
+        with gr.Row():
+
+            login_btn = gr.Button(
+                "Giriş Yap",
+                variant="primary"
+            )
+
+            signup_btn = gr.Button(
+                "Kayıt Ol"
+            )
+
+        auth_message = gr.Markdown()
+
+
+    # ========================================================
+    # APP
+    # ========================================================
+
+    with gr.Column(
+        visible=False
+    ) as app_panel:
+
+        with gr.Row():
+
+            user_status = gr.Markdown()
+
+            logout_btn = gr.Button(
+                "Çıkış"
+            )
+
+        gr.Markdown(
+            "## 📊 Crypto Sentiment Analysis"
+        )
+
+        text_input = gr.Textbox(
+            label="Kripto metni",
+            placeholder=(
+                "Bitcoin ETF inflows increased "
+                "sharply today..."
+            ),
+            lines=7
+        )
+
+        analyze_btn = gr.Button(
+            "🔍 Analiz Et",
+            variant="primary"
+        )
+
+        sentiment_output = gr.Textbox(
+            label="Sentiment"
+        )
+
+
+        # ====================================================
+        # ADMIN PANEL
+        # ====================================================
+
+        with gr.Column(
+            visible=False
+        ) as admin_panel:
+
+            gr.Markdown(
+                """
+# 👑 Admin Panel
+
+Kullanıcıları buradan yönetebilirsin.
+"""
+            )
+
+            users_table = gr.Dataframe(
+                headers=[
+                    "ID",
+                    "Email",
+                    "Plan",
+                    "Status",
+                    "Daily Usage",
+                    "Created"
+                ],
+                datatype=[
+                    "number",
+                    "str",
+                    "str",
+                    "str",
+                    "number",
+                    "str"
+                ],
+                value=get_users_dataframe(),
+                interactive=False
+            )
+
+            refresh_btn = gr.Button(
+                "🔄 Kullanıcıları Yenile"
+            )
+
+            gr.Markdown(
+                "### Üyelik Değiştir"
+            )
+
+            plan_user_id = gr.Number(
+                label="User ID"
+            )
+
+            plan_dropdown = gr.Dropdown(
+                choices=[
+                    "free",
+                    "pro",
+                    "admin"
+                ],
+                value="pro",
+                label="Yeni Plan"
+            )
+
+            change_plan_btn = gr.Button(
+                "Planı Değiştir"
+            )
+
+            admin_message = gr.Markdown()
+
+            gr.Markdown(
+                "### Kullanıcı Aktif / Pasif"
+            )
+
+            toggle_user_id = gr.Number(
+                label="User ID"
+            )
+
+            toggle_btn = gr.Button(
+                "Aktif / Pasif Değiştir"
+            )
+
+            gr.Markdown(
+                "### Kullanıcı Sil"
+            )
+
+            delete_user_id = gr.Number(
+                label="User ID"
+            )
+
+            delete_btn = gr.Button(
+                "🗑️ Kullanıcıyı Sil",
+                variant="stop"
+            )
+
+
+    # ========================================================
+    # EVENTS
+    # ========================================================
+
+    login_btn.click(
+        login,
+        inputs=[
+            email,
+            password
+        ],
+        outputs=[
+            auth_message,
+            session,
+            login_panel,
+            app_panel,
+            admin_panel,
+            password
+        ]
+    )
+
+    signup_btn.click(
+        signup,
+        inputs=[
+            email,
+            password
+        ],
+        outputs=[
+            auth_message
+        ]
+    )
+
+    logout_btn.click(
+        logout,
+        outputs=[
+            user_status,
+            session,
+            login_panel,
+            app_panel,
+            admin_panel,
+            sentiment_output
+        ]
+    )
+
+    analyze_btn.click(
+        analyze_crypto,
+        inputs=[
+            text_input,
+            session
+        ],
+        outputs=[
+            sentiment_output
+        ]
+    )
+
+    refresh_btn.click(
+        refresh_users,
+        outputs=[
+            users_table
+        ]
+    )
+
+    change_plan_btn.click(
+        admin_change_plan,
+        inputs=[
+            plan_user_id,
+            plan_dropdown
+        ],
+        outputs=[
+            admin_message
+        ]
+    )
+
+    toggle_btn.click(
+        admin_toggle_user,
+        inputs=[
+            toggle_user_id
+        ],
+        outputs=[
+            admin_message
+        ]
+    )
+
+    delete_btn.click(
+        admin_delete_user,
+        inputs=[
+            delete_user_id
+        ],
+        outputs=[
+            admin_message
+        ]
+    )
 
 
-# =========================
+# ============================================================
 # START
-# =========================
+# ============================================================
 
-demo.launch()
+demo.launch(
+    server_name="0.0.0.0",
+    server_port=7860
+)