testing

.idea/.gitignore

@@ -0,0 +1,8 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/DarkMuse.iml

@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$">
+      <excludeFolder url="file://$MODULE_DIR$/venv" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,13 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="Eslint" enabled="true" level="WARNING" enabled_by_default="true" />
+    <inspection_tool class="PyUnresolvedReferencesInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredIdentifiers">
+        <list>
+          <option value="str.*" />
+        </list>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/DarkMuse.iml" filepath="$PROJECT_DIR$/.idea/DarkMuse.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

app.py

@@ -3,6 +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 discord
 import requests
@@ -12,7 +14,7 @@ from discord.ext import commands
 from gradio_client import Client
 from gradio_client.exceptions import AppError
 
-# **Fetch Discord Bot Token from Environment Variable**
+# **Fetch Discord Bot Token and Hugging Face Token from Environment Variables**
 DISCORD_BOT_TOKEN = os.environ.get('DISCORD_BOT_TOKEN')
 HF_TOKEN = os.environ.get('HF_TOKEN')  # Fetch the HF_TOKEN
 
@@ -47,6 +49,9 @@ PROMPT_REGEX = re.compile(r'prompt\s*=\s*"(.*?)"', re.DOTALL)
 # Initialize the Gradio client with hf_token
 GRADIO_CLIENT = Client("Nevaehni/FLUX.1-schnell", hf_token=HF_TOKEN)
 
+# Initialize Hugging Face API client
+hf_api = HfApi()
+
 @bot.event
 async def on_ready():
     """Event handler triggered when the bot is ready."""
@@ -248,7 +253,32 @@ async def start_web_server():
 async def start_bot():
     await bot.start(DISCORD_BOT_TOKEN)
 
+async def schedule_restart():
+    """
+    Schedule the bot to restart the Hugging Face Space every day at 05:00.
+    """
+    space_name = "Nevaehni/FLUX.1-schnell"  # Replace with your actual space name
+    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)
+        try:
+            logger.info("Attempting to restart the Hugging Face Space...")
+            hf_api.restart_space(space_id=space_name, 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)
+
 async def main():
+    # Start the scheduler as a background task
+    asyncio.create_task(schedule_restart())
+    # Run the bot and web server concurrently
     await asyncio.gather(
         start_bot(),
         start_web_server()