Update app.py

app.py

@@ -5,9 +5,17 @@ import numpy as np
 from math import cos, radians, sin
 import os
 import time
+import json
 
-# Global variable to store generated GIFs
-generated_gifs = []
+def load_history():
+    try:
+        history = gr.get_state("generated_gifs")
+        return json.loads(history) if history else []
+    except:
+        return []
+
+def save_history(history):
+    gr.set_state("generated_gifs", json.dumps(history))
 
 def preprocess_image(image):
     # Ensure the image is square by padding it
@@ -103,12 +111,17 @@ def create_gif(editor1_output, editor2_output, transition_type):
         with open(output_path, "wb") as f:
             f.write(output.getvalue())
 
-        # Add the new GIF to the list of generated GIFs
-        global generated_gifs
-        generated_gifs.append((output_path, f"{transition_type.capitalize()} transition"))
+        # Load existing history
+        history = load_history()
+          
+        # Add the new GIF to the history
+        history.append((output_path, f"{transition_type.capitalize()} transition"))
+          
+        # Save updated history
+        save_history(history)
 
         # Return the updated list of generated GIFs
-        return generated_gifs
+        return history
 
     except Exception as e:
         raise ValueError(f"Error creating GIF: {e}")
@@ -162,5 +175,8 @@ with gr.Blocks() as iface:
         outputs=[output_gallery]
     )
 
+    # Load initial history when the app starts
+    iface.load(lambda: gr.update(value=load_history()), outputs=[output_gallery])
+
 # Launch the interface
 iface.launch(share=True)