Update app.py

app.py

@@ -6,6 +6,8 @@ import torch
 import random
 import numpy as np
 from datasets import Dataset
+from huggingface_hub import HfApi
+from datetime import datetime, time
 
 # Use spaces.gpu_device() to get the correct GPU device
 device = spaces.gpu_device()
@@ -20,6 +22,25 @@ sentiment_model = AutoModelForSequenceClassification.from_pretrained("distilbert
 # Global variables for feedback collection
 feedback_data = []
 
+# Initialize HfApi
+api = HfApi()
+
+def set_sleep_time():
+    # Set the Space to sleep between 2 AM and 6 AM UTC
+    sleep_start = time(hour=2, minute=0)
+    sleep_end = time(hour=6, minute=0)
+    
+    try:
+        api.set_space_sleep_time(
+            repo_id="Oranblock/Websitem",  # Replace with your actual Space name
+            sleep_start_time=sleep_start,
+            sleep_end_time=sleep_end,
+            timezone="UTC"
+        )
+        return "Sleep time set successfully"
+    except Exception as e:
+        return f"Error setting sleep time: {str(e)}"
+
 @spaces.GPU
 def generate_text():
     prompt = "This AI-driven website is unique because"
@@ -68,7 +89,14 @@ with gr.Blocks(theme=gr.themes.huggingface) as demo:
             feedback_button = gr.Button("Submit Feedback")
             feedback_output = gr.Textbox(label="Feedback Status")
     
+    sleep_button = gr.Button("Set Sleep Time")
+    sleep_output = gr.Textbox(label="Sleep Time Status")
+    
     generate_button.click(generate_content, outputs=[text_output, image_output, sentiment_output])
     feedback_button.click(save_feedback, inputs=[feedback_input, feedback_rating], outputs=feedback_output)
+    sleep_button.click(set_sleep_time, outputs=sleep_output)
+
+# Set sleep time when the app starts
+set_sleep_time()
 
 demo.launch()