Update main.py

main.py

@@ -19,18 +19,6 @@ app = FastAPI()
 API_URL = "https://api-inference.huggingface.co/models/ehristoforu/dalle-3-xl"
 API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
-def generate_image(prompt):
-    try:
-        payload = {"inputs": prompt}
-        response = requests.post(API_URL, headers=headers, json=payload)
-        response.raise_for_status()  # Raise an exception for HTTP errors
-        image_bytes = response.content
-        image = Image.open(io.BytesIO(image_bytes))
-        return image
-    except Exception as e:
-        # Handle the error gracefully, such as logging the error or returning a default image
-        raise gr.Error(f"Error generating image: {e}")
-        return "Experiencing high demand. Please retry shortly. Thank you for your patience"  # Return None or a default image in case of error
 
 
 
@@ -76,6 +64,23 @@ def send_it1(inputs, noise_level, proc=proc1):
     output = proc(prompt_with_noise)
     return output
     
+def generate_image(prompt_with_noise):
+    try:
+        global request_counter
+        request_counter += 1
+        timestamp = f"{time.time()}_{request_counter}"
+        prompt_with_noise = add_random_noise(inputs, noise_level) + f" - {timestamp}"
+        payload = {"inputs": prompt_with_noise}
+        response = requests.post(API_URL, headers=headers, json=payload)
+        response.raise_for_status()  # Raise an exception for HTTP errors
+        image_bytes = response.content
+        image = Image.open(io.BytesIO(image_bytes))
+        return image
+    except Exception as e:
+        # Handle the error gracefully, such as logging the error or returning a default image
+        raise gr.Error(f"Error generating image: {e}")
+        return "Experiencing high demand. Please retry shortly. Thank you for your patience"  # Return None or a default image in case of error
+   
 
 def get_prompts(prompt_text):
     if not prompt_text: