Update app.py

app.py

@@ -5,6 +5,26 @@ import random
 from PIL import Image
 import os
 import json
+import time
+
+# Restart mechanism: Check if the application has been running for more than 48 hours.
+start_time_file = "start_time.txt"
+if not os.path.exists(start_time_file):
+    # Create the file and write the current time (in seconds since epoch)
+    with open(start_time_file, "w") as f:
+        f.write(str(time.time()))
+else:
+    try:
+        with open(start_time_file, "r") as f:
+            start_time = float(f.read())
+        # Check if more than 48 hours have passed
+        if time.time() - start_time > 48 * 3600:
+            st.write("Restarting the space after 48 hours of uptime...")
+            os._exit(0)  # Force a restart of the space
+    except Exception as e:
+        # If there's an issue reading the file, reset the start time.
+        with open(start_time_file, "w") as f:
+            f.write(str(time.time()))
 
 # API Configuration
 API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
@@ -68,4 +88,4 @@ if prompt_from_url:
             mime="image/png"
         )
 else:
-    st.info("Add a 'text' parameter to the URL to generate an image. Example: ?text=astronaut+riding+a+horse")
+    st.info("Add a 'text' parameter to the URL to generate an image. Example: ?text=astronaut riding a horse")