Create app.py

app.py

@@ -0,0 +1,226 @@
+import gradio as gr
+import requests
+import json
+import re
+
+LLAMA_URL = "http://localhost:8080/v1/chat/completions"
+
+SYSTEM_PROMPT = """You are PromptForge, an AI that converts simple user statements into highly detailed JSON prompts for AI image and video generation.
+
+You MUST always respond with valid JSON only. No explanation, no extra text — just pure JSON.
+
+For IMAGE prompts, use this exact structure:
+{
+  "type": "image",
+  "prompt": "ultra-detailed visual description here",
+  "negative_prompt": "what to avoid",
+  "style": "e.g. photorealistic / anime / oil painting / cinematic",
+  "aspect_ratio": "e.g. 16:9 / 1:1 / 9:16",
+  "lighting": "e.g. golden hour / dramatic studio / soft diffused",
+  "camera": "e.g. wide angle / macro / aerial / 35mm portrait",
+  "mood": "e.g. mysterious / joyful / melancholic",
+  "quality_tags": ["8k", "ultra detailed", "award winning", "masterpiece"],
+  "color_palette": "e.g. warm earth tones / neon cyberpunk / pastel",
+  "additional_details": "any extra context"
+}
+
+For VIDEO prompts, use this exact structure:
+{
+  "type": "video",
+  "prompt": "ultra-detailed scene description",
+  "negative_prompt": "what to avoid",
+  "style": "e.g. cinematic / animation / documentary",
+  "aspect_ratio": "e.g. 16:9 / 9:16",
+  "duration_seconds": 5,
+  "fps": 24,
+  "motion": "e.g. slow zoom in / pan left / dolly shot / static",
+  "camera_movement": "e.g. handheld shaky / smooth gimbal / aerial drone",
+  "lighting": "e.g. natural daylight / neon lights / candlelit",
+  "mood": "e.g. tense / peaceful / epic",
+  "scene_transition": "e.g. fade in / cut / dissolve",
+  "audio_suggestion": "e.g. ambient rain / dramatic orchestral / none",
+  "quality_tags": ["cinematic 4k", "smooth motion", "color graded"],
+  "additional_details": "any extra context"
+}
+
+Always generate BOTH an image prompt AND a video prompt for every user input, combined into a single JSON object like this:
+{
+  "image": { ...image prompt object... },
+  "video": { ...video prompt object... }
+}
+
+Be creative, specific, and detailed. Expand on the user's simple statement richly."""
+
+
+def generate_prompts(user_input: str, history: list):
+    if not user_input.strip():
+        return history, history, ""
+
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+    for h in history:
+        messages.append({"role": "user", "content": h[0]})
+        messages.append({"role": "assistant", "content": h[1]})
+    messages.append({"role": "user", "content": user_input})
+
+    try:
+        response = requests.post(
+            LLAMA_URL,
+            json={
+                "model": "local",
+                "messages": messages,
+                "max_tokens": 1200,
+                "temperature": 0.7,
+                "stream": False,
+            },
+            timeout=120,
+        )
+        raw = response.json()["choices"][0]["message"]["content"].strip()
+
+        # Try to extract JSON from response
+        json_match = re.search(r'\{[\s\S]*\}', raw)
+        if json_match:
+            parsed = json.loads(json_match.group())
+            formatted = json.dumps(parsed, indent=2)
+            display = f"```json\n{formatted}\n```"
+        else:
+            display = raw
+
+        new_history = history + [(user_input, display)]
+        return new_history, new_history, ""
+
+    except Exception as e:
+        err = f"⚠️ Error: {str(e)}\n\nMake sure the llama.cpp server is running on port 8080."
+        new_history = history + [(user_input, err)]
+        return new_history, new_history, ""
+
+
+def clear_history():
+    return [], [], ""
+
+
+CSS = """
+@import url('https://fonts.googleapis.com/css2?family=Space+Mono:wght@400;700&family=DM+Sans:wght@300;400;600&display=swap');
+
+body, .gradio-container {
+    background: #0a0a0f !important;
+    color: #e8e8f0 !important;
+    font-family: 'DM Sans', sans-serif !important;
+}
+
+.gradio-container {
+    max-width: 900px !important;
+    margin: 0 auto !important;
+}
+
+h1 {
+    font-family: 'Space Mono', monospace !important;
+    font-size: 1.8rem !important;
+    color: #a78bfa !important;
+    text-align: center;
+    letter-spacing: -1px;
+    margin-bottom: 4px !important;
+}
+
+.subtitle {
+    text-align: center;
+    color: #6b6b8a;
+    font-size: 0.85rem;
+    margin-bottom: 24px;
+    font-family: 'Space Mono', monospace;
+}
+
+.chatbot .message {
+    font-family: 'Space Mono', monospace !important;
+    font-size: 0.78rem !important;
+}
+
+.chatbot .user {
+    background: #1a1a2e !important;
+    border: 1px solid #2d2d4e !important;
+    border-radius: 8px !important;
+}
+
+.chatbot .bot {
+    background: #0d1117 !important;
+    border: 1px solid #7c3aed44 !important;
+    border-radius: 8px !important;
+    color: #c4b5fd !important;
+}
+
+textarea, input {
+    background: #111120 !important;
+    border: 1px solid #2d2d4e !important;
+    color: #e8e8f0 !important;
+    border-radius: 8px !important;
+    font-family: 'DM Sans', sans-serif !important;
+}
+
+button.primary {
+    background: linear-gradient(135deg, #7c3aed, #a855f7) !important;
+    border: none !important;
+    border-radius: 8px !important;
+    color: white !important;
+    font-family: 'Space Mono', monospace !important;
+    font-weight: 700 !important;
+    letter-spacing: 1px !important;
+}
+
+button.secondary {
+    background: #1a1a2e !important;
+    border: 1px solid #2d2d4e !important;
+    color: #6b6b8a !important;
+    border-radius: 8px !important;
+}
+
+.examples-table td {
+    background: #111120 !important;
+    border: 1px solid #2d2d4e !important;
+    color: #a78bfa !important;
+    font-family: 'Space Mono', monospace !important;
+    font-size: 0.75rem !important;
+}
+"""
+
+with gr.Blocks(css=CSS, title="PromptForge") as demo:
+    gr.HTML("""
+        <h1>⚡ PromptForge</h1>
+        <p class="subtitle">simple words → detailed JSON image & video prompts</p>
+    """)
+
+    chatbot = gr.Chatbot(
+        label="",
+        height=480,
+        show_label=False,
+        render_markdown=True,
+        bubble_full_width=False,
+    )
+
+    state = gr.State([])
+
+    with gr.Row():
+        txt = gr.Textbox(
+            placeholder="Describe a scene, mood, or idea... e.g. 'a samurai in rainy Tokyo at night'",
+            show_label=False,
+            scale=5,
+            lines=1,
+        )
+        btn = gr.Button("FORGE →", variant="primary", scale=1)
+        clr = gr.Button("Clear", variant="secondary", scale=1)
+
+    gr.Examples(
+        examples=[
+            ["a lone astronaut on a red alien desert at sunset"],
+            ["a cozy Japanese cafe interior in winter morning"],
+            ["an ancient dragon flying over a burning medieval city"],
+            ["a neon-lit cyberpunk street market in heavy rain"],
+            ["a peaceful mountain lake reflection at golden hour"],
+        ],
+        inputs=txt,
+        label="Try these →",
+    )
+
+    btn.click(generate_prompts, [txt, state], [chatbot, state, txt])
+    txt.submit(generate_prompts, [txt, state], [chatbot, state, txt])
+    clr.click(clear_history, [], [chatbot, state, txt])
+
+demo.launch(server_name="0.0.0.0", server_port=7860)