Update app.py

app.py

@@ -95,10 +95,9 @@ def generate_prompt(image, api_key, style, creativity, neg_prompt, aspect, color
         input_tokens = len(img_b64) // 4  # Approximate
         output_tokens = len(raw_prompt.split())
 
-        return {
-            "prompt": raw_prompt,
-            "validation": validation,
-            "stats": {"input": input_tokens, "output": output_tokens}
+        return raw_prompt, validation, {
+            "input": input_tokens,
+            "output": output_tokens
         }
 
     except Exception as e: