import gradio as gr

# -------------------------
# SMART UNDERSTANDING LOGIC
# -------------------------

def detect_mode(user_text):
    text = user_text.lower()

    if any(k in text for k in ["upsc", "exam", "notes", "school", "college"]):
        return "academic"
    if any(k in text for k in ["instagram", "facebook", "dp", "profile", "post"]):
        return "social"
    if any(k in text for k in ["diagram", "flow", "map"]):
        return "diagram"
    return "general"


def build_prompt(user_text):
    mode = detect_mode(user_text)

    base_realism = (
        "highly realistic, photorealistic image, "
        "natural lighting, real human texture, "
        "no cartoon, no fantasy, professional quality"
    )

    if mode == "academic":
        return (
            f"{base_realism}, academic textbook style, "
            f"exam-safe visual, subject based on: {user_text}"
        )

    if mode == "social":
        return (
            f"{base_realism}, premium social media portrait, "
            f"clean background, instagram facebook ready, "
            f"based on: {user_text}"
        )

    if mode == "diagram":
        return (
            f"clean academic diagram, simple design, "
            f"clear labels, NCERT style, topic: {user_text}"
        )

    return f"{base_realism}, based on: {user_text}"


NEGATIVE_PROMPT = (
    "cartoon, anime, illustration, painting, fantasy, unreal, "
    "plastic skin, blurry, distorted, watermark, text, logo"
)

# -------------------------
# IMAGE GENERATION FUNCTION
# -------------------------
def generate_image(user_text):
    final_prompt = build_prompt(user_text)

    # 🔴 यहां अपना MODEL / API call लगाना है
    # image = model.generate(
    #     prompt=final_prompt,
    #     negative_prompt=NEGATIVE_PROMPT
    # )
    # return image

    return None  # placeholder (UI test के लिए)

# -------------------------
# GRADIO UI
# -------------------------
with gr.Blocks(title="Smart Text-to-Art AI") as demo:
    gr.Markdown(
        """
        ## 🎓 Smart Text-to-Art AI Generator  
        बस लिखो, AI खुद समझेगा और realistic image बनाएगा  

        ✔ Student Safe  
        ✔ Social Media Ready  
        ✔ No Cartoon / No Fantasy  
        """
    )

    user_input = gr.Textbox(
        label="आप क्या बनवाना चाहते हैं?",
        placeholder="Example: UPSC ke liye raja ki realistic photo"
    )

    generate_btn = gr.Button("Generate Image")
    output = gr.Image(label="Output", type="pil")

    generate_btn.click(
        fn=generate_image,
        inputs=user_input,
        outputs=output
    )

if __name__ == "__main__":
    demo.launch()