Update app.py

app.py

@@ -3,67 +3,87 @@ import requests
 import json
 import re
 
-# llama.cpp server endpoint
 LLAMA_URL = "http://localhost:8080/v1/chat/completions"
 
-SYSTEM_PROMPT = """You are PromptForge. Convert user input into detailed JSON prompts. 
-Respond ONLY with a single JSON object. No conversational filler.
+# Advanced System Prompt: Prevents repetition and forces layered details
+SYSTEM_PROMPT = """You are PromptForge. Create hyper-detailed JSON prompts.
+STRICT RULE: Never repeat a phrase or concept within the same prompt. 
+Each sentence must provide NEW information (texture, lighting, specific movement, or camera angle).
 
-Format:
+If 'Image' is active:
+- Focus on the 'frozen moment': materials (brushed steel, translucent skin), 
+  lighting (crepuscular rays, neon flicker), and composition.
+
+If 'Video' is active:
+- Focus on 'progression': How does the light change? What is the specific 
+  physics of the motion? (e.g., 'the dragon's weight displaces the forest canopy').
+
+Response must be ONLY valid JSON:
 {
-  "image": {
-    "prompt": "detailed visual description",
-    "style": "photorealistic/anime/etc",
-    "aspect_ratio": "16:9",
-    "lighting": "description"
-  },
-  "video": {
-    "prompt": "scene description with motion",
-    "motion": "camera movement details",
-    "duration": 5
-  }
+  "image": { "prompt": "...", "style": "...", "aspect_ratio": "...", "lighting": "..." },
+  "video": { "prompt": "...", "motion": "...", "duration": 5 }
 }"""
 
-def generate(user_input):
-    if not user_input.strip():
-        return None
+def generate(user_input, selected_modes):
+    if not user_input.strip() or not selected_modes:
+        return {"error": "Input text and at least one mode (Image/Video) required."}
 
-    # "One and Done" - We only send the system prompt and the current message
+    # Instruct the model on which keys to generate
+    mode_context = f"Generate only for: {', '.join(selected_modes)}."
+    
     payload = {
         "messages": [
             {"role": "system", "content": SYSTEM_PROMPT},
-            {"role": "user", "content": user_input}
+            {"role": "user", "content": f"Task: {user_input}\nContext: {mode_context}\nRequirement: High complexity, zero repetition, extreme sensory detail."}
         ],
-        "temperature": 0.2, # Low temp for strict JSON
-        "max_tokens": 1000
+        "temperature": 0.45, # Slightly higher for creative vocabulary
+        "max_tokens": 1200
     }
 
     try:
-        response = requests.post(LLAMA_URL, json=payload, timeout=60)
+        response = requests.post(LLAMA_URL, json=payload, timeout=90)
         content = response.json()["choices"][0]["message"]["content"].strip()
         
-        # Clean up the output to ensure it's just the JSON block
+        # Robust JSON extraction
         json_match = re.search(r'\{[\s\S]*\}', content)
         if json_match:
             return json.loads(json_match.group())
-        return {"error": "Model failed to return valid JSON", "raw": content}
+        return {"error": "Model output was not valid JSON.", "raw_content": content}
     except Exception as e:
-        return {"error": str(e)}
+        return {"error": f"Connection Error: {str(e)}"}
 
-# Dark-themed UI for a clean "Forge" feel
+# --- UI Setup ---
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple", neutral_hue="slate")) as demo:
-    gr.Markdown("# ⚡ PromptForge: JSON Generator")
-    gr.Markdown("Input a simple idea to get structured Image & Video prompts.")
+    gr.HTML("<h1 style='text-align: center; margin-bottom: 0;'>⚡ PromptForge</h1>")
+    gr.HTML("<p style='text-align: center; color: #888;'>Press <b>Enter</b> or click <b>Forge</b> to generate</p>")
     
     with gr.Row():
         with gr.Column(scale=1):
-            input_text = gr.Textbox(label="Idea", placeholder="e.g. A cybernetic dragon in a neon forest")
-            run_btn = gr.Button("Generate Prompts", variant="primary")
-        
+            input_text = gr.Textbox(
+                label="Core Idea", 
+                placeholder="e.g. A cybernetic dragon in a forest...",
+                lines=3,
+                autofocus=True # Sets focus so you can type immediately
+            )
+            
+            # Selection for Image/Video
+            modes = gr.CheckboxGroup(
+                choices=["Image", "Video"], 
+                value=["Image", "Video"], 
+                label="Prompt Types to Generate"
+            )
+            
+            run_btn = gr.Button("FORGE →", variant="primary")
+            
         with gr.Column(scale=2):
-            output_json = gr.JSON(label="Generated JSON Prompts")
+            output_json = gr.JSON(label="Resulting JSON Data")
 
-    run_btn.click(fn=generate, inputs=input_text, outputs=output_json)
+    # --- THE MAGIC TRIGGERS ---
+    # 1. Clicking the button
+    run_btn.click(fn=generate, inputs=[input_text, modes], outputs=output_json)
+    
+    # 2. Hitting ENTER on the keyboard
+    input_text.submit(fn=generate, inputs=[input_text, modes], outputs=output_json)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)