testing

.idea/DarkMuse.iml

@@ -7,4 +7,8 @@
     <orderEntry type="inheritedJdk" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
+  <component name="PyDocumentationSettings">
+    <option name="format" value="PLAIN" />
+    <option name="myDocStringFormat" value="Plain" />
+  </component>
 </module>

.idea/misc.xml

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Black">
+    <option name="sdkName" value="Python 3.10 (DarkMuse)" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10 (DarkMuse)" project-jdk-type="Python SDK" />
+</project>

app.py

@@ -3,8 +3,8 @@ import logging
 import os
 import re
 import sys
-import datetime  # Added for scheduling
-from huggingface_hub import HfApi  # Added for Hugging Face API
+import datetime  # For scheduling
+from huggingface_hub import HfApi  # For Hugging Face API
 
 import discord
 import requests
@@ -255,25 +255,23 @@ async def start_bot():
 
 async def schedule_restart():
     """
-    Schedule the bot to restart the Hugging Face Space every day at 05:00.
+    Schedule the bot to restart the Hugging Face Space at a specified interval.
+    For testing purposes, this function restarts the Space every minute.
     """
-    space_name = "Nevaehni/FLUX.1-schnell"  # Replace with your actual space name
+    space_id = "Nevaehni/FLUX.1-schnell"  # Replace with your actual space ID
+    restart_interval_seconds = 60  # Restart every 60 seconds (1 minute) for testing
+
     while True:
-        now = datetime.datetime.now()
-        target_time = now.replace(hour=5, minute=0, second=0, microsecond=0)
-        if now >= target_time:
-            target_time += datetime.timedelta(days=1)
-        wait_seconds = (target_time - now).total_seconds()
-        logger.info(f"Scheduled space restart in {wait_seconds} seconds at {target_time.strftime('%Y-%m-%d %H:%M:%S')}.")
-        await asyncio.sleep(wait_seconds)
+        logger.info(f"Scheduled space restart in {restart_interval_seconds} seconds.")
+        await asyncio.sleep(restart_interval_seconds)
         try:
             logger.info("Attempting to restart the Hugging Face Space...")
-            hf_api.restart_space(space_id=space_name, token=HF_TOKEN)
+            hf_api.restart_space(space_id=space_id, token=HF_TOKEN)
             logger.info("Space restarted successfully.")
         except Exception as e:
             logger.error(f"Failed to restart space: {e}")
         # Optional: Add a short delay to prevent rapid retries in case of failure
-        await asyncio.sleep(60)
+        await asyncio.sleep(5)  # Wait 5 seconds before the next restart cycle
 
 async def main():
     # Start the scheduler as a background task