Create app.py

app.py

@@ -0,0 +1,88 @@
+import os
+import random
+from smolagents import CodeAgent, InferenceClientModel
+import gradio as gr
+
+HF_TOKEN = os.getenv("HF_TOKEN")
+model_id = "mistralai/Mistral-7B-Instruct-v0.1"
+
+client_model = InferenceClientModel(
+    model_id=model_id,
+    token=HF_TOKEN,
+)
+
+agent = CodeAgent(
+    llm=client_model,
+    system_message="You are a prompt generator specialized in hyper-realistic cosplay portrait prompts."
+)
+
+CHARACTER_FEMALES = [
+    {"name": "Rias Gremory", "series": "High School DxD", "description": "crimson hair styled in long waves, school uniform with a short skirt", "eye_color": "blue", "hair_color": "crimson"},
+    {"name": "Tifa Lockhart", "series": "Final Fantasy VII", "description": "black hair in ponytail, sporty tank top and mini skirt", "eye_color": "brown", "hair_color": "black"},
+    {"name": "Mikasa Ackerman", "series": "Attack on Titan", "description": "short black hair, scout uniform, fitted jacket and pants", "eye_color": "gray", "hair_color": "black"},
+    {"name": "Sailor Moon", "series": "Sailor Moon", "description": "long blonde hair in odango, sailor scout uniform with pleated skirt", "eye_color": "blue", "hair_color": "blonde"},
+    {"name": "Hinata Hyuga", "series": "Naruto", "description": "dark blue hair, ninja outfit with lavender jacket and pants", "eye_color": "pale lavender", "hair_color": "dark blue"}
+]
+
+SETTINGS = [
+    "neon-lit office at midnight with hum of computers and scent of rain",
+    "gothic library filled with flickering candlelight and faded parchment scent",
+    "throne room bathed in warm golden light with velvet drapes",
+    "sterile operating room with bright surgical lights and soft beeping sounds",
+    "magical forest with glowing flora and gentle breeze rustling leaves"
+]
+
+ACTIVITIES = [
+    "reaching for a top-shelf file",
+    "arranging occult tomes on a desk",
+    "casting a subtle spell with glowing magical effects",
+    "serving coffee with graceful movements",
+    "walking confidently down a gothic hallway"
+]
+
+LINGERIE_COLORS = ["black", "red", "white", "navy blue", "deep burgundy"]
+
+STOCKING_DETAILS = ["a delicate back seam", "intricate lace pattern", "classic fishnet design", "smooth satin finish", "subtle floral embroidery"]
+
+HEEL_DETAILS = ["sharp stiletto heels", "elegant kitten heels", "strappy high heels", "sleek patent leather pumps", "classic pointed toe heels"]
+
+POSES = ["confident stance, one hand on hip", "sitting casually on a desk", "leaning slightly forward", "standing with arms crossed", "walking with a playful smile"]
+
+EXPRESSIONS = ["a knowing, seductive smile", "a confident, alluring gaze", "a playful smirk", "an intense, focused look", "a soft, inviting smile"]
+
+def generate_prompt(character, setting, activity, lingerie_color, stocking_detail, heel_detail, pose, expression, prompt_num):
+    return f"""### Prompt {prompt_num}
+**{character['name']}** (*{character['series']}*) is captured in a professional Ultra HD 16K (15360x8640) RAW photograph, set in **{setting}**, the ambiance filled with sensory elements. Her flawless cosplay features **{character['hair_color']} hair styled {character['description']}**, accented by **{character['eye_color']} eyes**, cascading over a **{lingerie_color} lace thong** and matching transparent lace bra. **Thigh-high stockings** with **{stocking_detail}** accentuate her legs, paired with **{heel_detail}**. The character is posed in a **{pose}**, her **{expression}** directed at the viewer. Shot from a **low-angle (worm’s eye view)** using a **Canon EOS R5 Cine RAW** with a **Canon RF 85mm f/1.2L USM lens**, the composition emphasizes her dominance and the sensual details of her lingerie. **ARRI SkyPanel S360-C lighting** creates soft, flattering illumination, enhancing realistic skin texture and fabric sheen, while **DaVinci Resolve color grading** adds vibrant, cinematic tones. The full vertical frame (knees to head) ensures no cropping, with perfect depth of field and subtle bokeh background.
+Negative prompt keywords exclude cartoon, blurry, pixelated, low-resolution, watermark, noise, overexposed, underexposed, unnatural shadows, color banding, oversaturation, artificial textures, and all other disallowed artifacts."""
+
+def chat_with_agent(user_message):
+    if user_message.strip().lower() == "ok":
+        prompts = []
+        selected_chars = random.sample(CHARACTER_FEMALES, 5)
+        for i, char in enumerate(selected_chars, start=1):
+            prompt = generate_prompt(
+                char,
+                random.choice(SETTINGS),
+                random.choice(ACTIVITIES),
+                random.choice(LINGERIE_COLORS),
+                random.choice(STOCKING_DETAILS),
+                random.choice(HEEL_DETAILS),
+                random.choice(POSES),
+                random.choice(EXPRESSIONS),
+                i
+            )
+            prompts.append(prompt)
+        return "\n\n".join(prompts)
+    else:
+        return "Please enter 'Ok' to generate 5 detailed cosplay prompts."
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Automatic Hyper-Realistic Cosplay Prompt Generator")
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(label="Type 'Ok' to generate 5 automatic prompts")
+    msg.submit(chat_with_agent, msg, chatbot)
+    msg.submit(lambda: "", None, msg)  # Clear input after submit
+
+if __name__ == "__main__":
+    demo.launch()
+