Update main.py

main.py

@@ -64,23 +64,28 @@ def send_it1(inputs, noise_level, proc=proc1):
     output = proc(prompt_with_noise)
     return output
     
-def generate_image(prompt_with_noise):
+def generate_image(inputs):
     try:
         global request_counter
         request_counter += 1
         timestamp = f"{time.time()}_{request_counter}"
-        prompt_with_noise = add_random_noise(inputs, noise_level) + f" - {timestamp}"
+        prompt_with_noise = add_random_noise(inputs) + 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 requests.exceptions.HTTPError as e:
+        # Handle any HTTP errors
+        print(f"HTTP Error: {e}")
+        return "An unexpected error occurred while generating the 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
-   
+        # Handle other exceptions
+        print(f"Error generating image: {e}")
+        return "An unexpected error occurred while generating the image. Please try again later."
+
+
 
 def get_prompts(prompt_text):
     if not prompt_text: