Update app.py

app.py

@@ -40,9 +40,13 @@ def authenticate_user(username, password):
 def list_saved_images(): 
     return sorted(DATA_PATH.glob("*.jpg"), key=lambda x: x.stat().st_mtime, reverse=True)
 
-def enhance_prompt(text, client=client):
+def enhance_prompt(text, client=client, use_enhancement=True):
+    if not use_enhancement:
+        return text[:200]
+    
     if not client:
         return text[:200]
+    
     try:
         enhanced = client.text_generation(
             "Generate a photorealistic, detailed txt2img prompt: " + text,
@@ -53,10 +57,6 @@ def enhance_prompt(text, client=client):
         st.warning(f"Prompt enhancement error: {e}")
         return text[:200]
 
-def save_prompt(image_name, enhanced_prompt):
-    with open(DATA_PATH / "prompts.txt", "a") as f: 
-        f.write(f"{image_name}: {enhanced_prompt}\n")
-
 def generate_variations(prompt, num_variants=8, use_enhanced=True):
     instructions = [
         "Photorealistic description for txt2img prompt: ",