Update app.py

app.py

@@ -408,7 +408,7 @@ def generate_high_quality_image(prompt, model_choice="dreamshaper-8", style="chi
     """Legacy function for backward compatibility"""
     return generate_consistent_image(prompt, model_choice, style, [], {}, 1)
 
-# LOCAL FILE MANAGEMENT FUNCTIONS (unchanged)
+# LOCAL FILE MANAGEMENT FUNCTIONS
 def save_image_to_local(image, prompt, style="test"):
     """Save image to local persistent storage"""
     try:
@@ -488,7 +488,7 @@ def refresh_local_images():
         print(f"Error refreshing local images: {e}")
         return []
 
-# OCI BUCKET FUNCTIONS (unchanged)
+# OCI BUCKET FUNCTIONS
 def save_to_oci_bucket(image, text_content, story_title, page_number, file_type="image"):
     """Save both images and text to OCI bucket via your OCI API"""
     try:
@@ -887,6 +887,38 @@ async def delete_local_image_api(filename: str):
     except Exception as e:
         return {"status": "error", "message": str(e)}
 
+# MISSING HELPER FUNCTIONS FOR GRADIO INTERFACE
+def delete_current_image(filepath):
+    """Delete the currently displayed image"""
+    if not filepath:
+        return "❌ No image to delete", None, None, refresh_local_images()
+    
+    success, message = delete_local_image(filepath)
+    updated_files = refresh_local_images()
+    
+    if success:
+        status_msg = f"✅ {message}"
+        return status_msg, None, "Image deleted successfully!", updated_files
+    else:
+        return f"❌ {message}", None, "Delete failed", updated_files
+
+def clear_all_images():
+    """Delete all local images"""
+    try:
+        storage_info = get_local_storage_info()
+        deleted_count = 0
+        
+        if "images" in storage_info:
+            for image_info in storage_info["images"]:
+                success, _ = delete_local_image(image_info["path"])
+                if success:
+                    deleted_count += 1
+        
+        updated_files = refresh_local_images()
+        return f"✅ Deleted {deleted_count} images", updated_files
+    except Exception as e:
+        return f"❌ Error: {str(e)}", refresh_local_images()
+
 # Enhanced Gradio interface with dynamic character testing
 def create_gradio_interface():
     """Create Gradio interface with dynamic character consistency features"""