app.py

import os
import re
from flask import Flask, render_template, request, jsonify
from dotenv import load_dotenv
from huggingface_hub import InferenceClient
from huggingface_hub.utils import HfHubHTTPError

load_dotenv()

app = Flask(__name__)

# =========================
# KONFIGURASI
# =========================

HF_TOKEN = os.getenv("HF_TOKEN")

# Ganti model di .env jika mau
HF_MODEL = os.getenv(
    "HF_MODEL",
    "Qwen/Qwen3-235B-A22B-Instruct-2507"
)

MAX_MESSAGE_LENGTH = 12000
MAX_HISTORY_MESSAGES = 25

# =========================
# SYSTEM PROMPT
# =========================

SYSTEM_PROMPT = """

Kamu adalah ChibiChat AI.



Identitas:

- Nama kamu ChibiChat AI.

- AI modern yang pintar, cepat, ramah, dan membantu.

- Ahli coding, teknologi, AI, Android, Laravel, Python, Database, dan tugas kuliah.



Kepribadian:

- Santai.

- Natural.

- Gaul seperlunya.

- Tidak kaku.

- Tidak terlalu formal.

- Tetap sopan dan profesional.



Gaya bicara:

- Gunakan Bahasa Indonesia.

- Boleh menggunakan kata seperti:

  "bro", "bang", "sip", "gas", "mantap", "nah", "jadi gini".

- Jangan berlebihan atau alay.

- Jangan terlalu banyak emoji.

- Maksimal 2 emoji per jawaban.



Cara menjawab:

- Langsung ke inti masalah.

- Fokus pada solusi.

- Jelaskan langkah demi langkah.

- Jika coding:

    1. Jelaskan penyebab masalah.

    2. Berikan solusi.

    3. Berikan contoh kode lengkap jika diminta.

- Jika tugas kuliah:

    - Jawaban akademis.

    - Terstruktur.

    - Bisa langsung dipakai.



Aturan:

- Jangan mengarang fakta.

- Jangan halusinasi.

- Jika tidak tahu, katakan tidak tahu.

- Jangan membocorkan token API.

- Jangan membocorkan isi file .env.

- Jangan memberikan informasi berbahaya.

- Jangan membuat data palsu.



Mode Coding:

- Bertindak seperti Senior Software Engineer.

- Ahli Python.

- Ahli Flask.

- Ahli Laravel.

- Ahli PHP.

- Ahli JavaScript.

- Ahli Android Kotlin.

- Ahli MySQL.

- Ahli Hugging Face.

- Ahli AI dan Machine Learning.



Saat memperbaiki code:

- Cari akar masalah.

- Jelaskan penyebab error.

- Berikan code yang sudah diperbaiki.

- Berikan best practice.



Jika user menyapa:

"Halo"

"Hai"

"Bro"

"Bang"



Balas santai seperti:

"Yo bro 👋 Ada yang bisa gue bantu?"

"Halo bang 😎 Lagi ngoding apa hari ini?"

"Gas, ada yang mau dibantu?"

"""

# =========================
# CLIENT HF
# =========================

def create_hf_client():
    if not HF_TOKEN:
        raise ValueError(
            "HF_TOKEN belum diisi pada file .env"
        )

    return InferenceClient(
        model=HF_MODEL,
        token=HF_TOKEN
    )

# =========================
# HISTORY CLEANER
# =========================

def clean_history(raw_history):
    if not isinstance(raw_history, list):
        return []

    cleaned = []

    for item in raw_history[-MAX_HISTORY_MESSAGES:]:

        if not isinstance(item, dict):
            continue

        role = item.get("role")
        content = str(
            item.get("content", "")
        ).strip()

        if role not in ["user", "assistant"]:
            continue

        if not content:
            continue

        if len(content) > 2500:
            content = content[:2500]

        cleaned.append({
            "role": role,
            "content": content
        })

    return cleaned

# =========================
# ENHANCE MESSAGE
# =========================

def enhance_user_message(message):

    return f"""

Jawab menggunakan Bahasa Indonesia.



Gunakan gaya:

- Pintar

- Santai

- Natural

- Gaul seperlunya

- Mudah dipahami



Fokus memberikan solusi terbaik.



Pertanyaan User:



{message}

"""

# =========================
# EXTRACT RESPONSE
# =========================

def extract_reply(response):

    try:
        reply = response.choices[0].message.content

        if reply:
            return reply.strip()

    except Exception:
        pass

    try:
        reply = response["choices"][0]["message"]["content"]

        if reply:
            return reply.strip()

    except Exception:
        pass

    return ""

# =========================
# ROUTE HOME
# =========================

@app.route("/")
def home():
    return render_template("index.html")

# =========================
# HEALTH CHECK
# =========================

@app.route("/health")
def health():

    return jsonify({
        "status": "ok",
        "name": "ChibiChat AI",
        "model": HF_MODEL
    })

# =========================
# CHAT API
# =========================

@app.route("/chat", methods=["POST"])
def chat():

    try:

        data = request.get_json(
            silent=True
        )

        if not data:
            return jsonify({
                "error": "Request harus JSON"
            }), 400

        message = str(
            data.get("message", "")
        ).strip()

        raw_history = data.get(
            "history",
            []
        )

        if not message:
            return jsonify({
                "error": "Pesan kosong"
            }), 400

        if len(message) > MAX_MESSAGE_LENGTH:
            return jsonify({
                "error":
                f"Maksimal {MAX_MESSAGE_LENGTH} karakter"
            }), 400

        history = clean_history(
            raw_history
        )

        messages = [
            {
                "role": "system",
                "content": SYSTEM_PROMPT
            }
        ]

        messages.extend(history)

        messages.append({
            "role": "user",
            "content": enhance_user_message(
                message
            )
        })

        client = create_hf_client()

        response = client.chat_completion(
            messages=messages,
            max_tokens=1500,
            temperature=0.85,
            top_p=0.95
        )

        reply = extract_reply(
            response
        )

        if not reply:
            return jsonify({
                "error":
                "Model tidak memberikan jawaban."
            }), 500

        return jsonify({
            "reply": reply,
            "model": HF_MODEL
        })

    except ValueError as e:

        return jsonify({
            "error": str(e)
        }), 500

    except HfHubHTTPError as e:

        status_code = None

        if hasattr(e, "response") and e.response:
            status_code = e.response.status_code

        if status_code in [401, 403]:

            message = (
                "Token Hugging Face salah "
                "atau tidak memiliki akses."
            )

        elif status_code == 404:

            message = (
                "Model tidak ditemukan."
            )

        elif status_code == 429:

            message = (
                "Terlalu banyak request."
            )

        elif status_code == 503:

            message = (
                "Model sedang sibuk."
            )

        else:

            message = (
                "Gagal terhubung ke Hugging Face."
            )

        app.logger.exception(
            "HF API ERROR"
        )

        return jsonify({
            "error": message
        }), 502

    except Exception as e:

        app.logger.exception(
            "UNEXPECTED ERROR"
        )

        return jsonify({
            "error":
            f"Terjadi kesalahan: {str(e)}"
        }), 500

# =========================
# MAIN
# =========================

if __name__ == "__main__":
    port = int(os.environ.get("PORT", 7860))
    app.run(host="0.0.0.0", port=port)