Update app.py

app.py

@@ -3,8 +3,12 @@ import requests
 import base64
 import json
 import os
+import logging
 from time import sleep
 
+# Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
 def file_to_base64(filepath):
     with open(filepath, "rb") as file:
         file_data = file.read()
@@ -13,9 +17,18 @@ def file_to_base64(filepath):
     return base64_message
 
 def submit_job(audio_file, preset, beat_sensitivity, fps, width, height):
+    # Check file size
+    file_size_mb = os.path.getsize(audio_file.name) / (1024 * 1024)
+    if file_size_mb > 3.5:
+        error_message = f"Error: File size is {file_size_mb:.2f}MB, which exceeds the 3.5MB limit."
+        logging.error(error_message)
+        return error_message, None
+
     api_key = os.getenv('DEFORUM_API_KEY')  # Ensure your environment variable is set
     if not api_key:
-        return "Error: API key is not set in environment variables", None
+        error_message = "Error: API key is not set in environment variables"
+        logging.error(error_message)
+        return error_message, None
 
     server = "https://deforum.studio"
     api_url = f'{server}/api/public/v1/audiovis1'
@@ -25,7 +38,6 @@ def submit_job(audio_file, preset, beat_sensitivity, fps, width, height):
     }
 
     audio_base64 = file_to_base64(audio_file.name)
-
     payload = {
         "audioData": audio_base64,
         "presetName": preset,
@@ -37,32 +49,42 @@ def submit_job(audio_file, preset, beat_sensitivity, fps, width, height):
 
     response = requests.post(api_url, headers=headers, json=payload)
     if response.status_code != 201:
-        return f"Error submitting job: {response.status_code} - {response.text}", None
+        error_message = f"Error submitting job: {response.status_code} - {response.text}"
+        logging.error(error_message)
+        return error_message, None
 
     data = response.json()
     tracking_url = f"{server}{data['links']['audiovis1']}"
+    logging.info(f"Job submitted successfully. Tracking URL: {tracking_url}")
 
     while True:
         response = requests.get(tracking_url, headers=headers)
         if response.status_code != 200:
-            return f"Error getting job status: {response.text}", None
+            error_message = f"Error getting job status: {response.text}"
+            logging.error(error_message)
+            return error_message, None
 
         tracking_data = response.json()
+        logging.info(f"Job status: {tracking_data['status']}")
         if tracking_data['status'] in ['canceled', 'failed', 'succeeded']:
             break
         sleep(10)
 
     if tracking_data['status'] != 'succeeded':
-        return f"Job ended with status: {tracking_data['status']}", None
+        error_message = f"Job ended with status: {tracking_data['status']}"
+        logging.error(error_message)
+        return error_message, None
     else:
-        return tracking_data['links']['outputUrls'][0]
+        output_url = tracking_data['links']['outputUrls'][0]
+        logging.info(f"Job succeeded. Output URL: {output_url}")
+        return output_url
 
 description1 = """
 # Audio Visualizer Playground
-### Easily create audio-synced animation masks for AnimateDiff or Deforum Animations.
+### A way for you to easily create audio-synced animation masks for AnimateDiff or Deforum Animations.
 """
 description2 = """
-#### Please provide feedback for us about which masks you would like to see in our Discord: [discord.gg/deforum](https://discord.gg/deforum)
+#### Please provide feedback for us about which masks you like, or would like to see in our animation channel in our Discord: [discord.gg/deforum](https://discord.gg/deforum)
 """
 
 custom_css = """
@@ -158,7 +180,7 @@ with gr.Blocks(css=custom_css) as demo:
                 "pulsing-circles", "pulsing-hexagons", "pulsing-squares", "pulsing-triangles", "rotating-shapes"
             ])
         with gr.Row():
-            beat_sensitivity = gr.Slider(label="Beat Sensitivity", minimum=1, maximum=5, step=0.1, value=4)
+            beat_sensitivity = gr.Slider(label="Beat Sensitivity", minimum=1, maximum=5, step=0.1, value=3)
             fps = gr.Slider(label="FPS", minimum=12, maximum=24, step=1, value=24)
         with gr.Row():
             width = gr.Slider(label="Width", minimum=512, maximum=1024, step=1, value=512)