Update app.py

app.py

@@ -12,8 +12,8 @@ def file_to_base64(filepath):
         base64_message = base64_encoded_data.decode('utf-8')
     return base64_message
 
-def submit_job(audio_file, preset, beat_sensitivity, fps):
-    api_key = os.getenv('DEFORUM_API_KEY')  
+def submit_job(audio_file, preset, beat_sensitivity, fps, width, height):
+    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
 
@@ -30,7 +30,9 @@ def submit_job(audio_file, preset, beat_sensitivity, fps):
         "audioData": audio_base64,
         "presetName": preset,
         "beatSensitivity": beat_sensitivity,
-        "fps": fps
+        "fps": fps,
+        "width": width,
+        "height": height
     }
 
     response = requests.post(api_url, headers=headers, json=payload)
@@ -94,7 +96,7 @@ h1, h2, h3, h4, h5, h6 {
 
 label {
     font-weight: bold;
-    color: #bbbbbb;
+    color: #000000; /* Change label color to black */
     text-align: center;
 }
 
@@ -142,27 +144,33 @@ video, .gr-video {
 }
 """
 
-def main(audio_file, preset, beat_sensitivity, fps):
-    video_url = submit_job(audio_file, preset, beat_sensitivity, fps)
+def main(audio_file, preset, beat_sensitivity, fps, width, height):
+    video_url = submit_job(audio_file, preset, beat_sensitivity, fps, width, height)
     return video_url
 
 with gr.Blocks(css=custom_css) as demo:
     gr.Markdown(description1, elem_id="markdown-text")
     gr.Markdown(description2, elem_id="markdown-text")
-    audio_file = gr.File(label="Upload MP3 File", file_types=['audio'])
-    preset = gr.Dropdown(label="Select Visualization Preset", choices=[
-        "pulsing-circles", "pulsing-hexagons", "pulsing-squares", "pulsing-triangles", 
-        "rewbs-01", "rewbs-02", "rotating-shapes"
-    ])
-    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)
-    submit_button = gr.Button("Submit")
+    with gr.Group():
+        with gr.Column():
+            audio_file = gr.File(label="Upload MP3 File", file_types=['audio'])
+            preset = gr.Dropdown(label="Select Visualization Preset", choices=[
+                "pulsing-circles", "pulsing-hexagons", "pulsing-squares", "pulsing-triangles", 
+                "rewbs-01", "rewbs-02", "rotating-shapes"
+            ])
+        with gr.Row():
+            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)
+            height = gr.Slider(label="Height", minimum=512, maximum=1024, step=1, value=512)
+        submit_button = gr.Button("Submit")
     output_video = gr.Video(label="Output MP4")
 
     def update_output(video_url):
         output_video.update(value=video_url, visible=True)
 
-    submit_button.click(main, inputs=[audio_file, preset, beat_sensitivity, fps], outputs=[output_video])
+    submit_button.click(main, inputs=[audio_file, preset, beat_sensitivity, fps, width, height], outputs=[output_video])
 
 if __name__ == "__main__":
     demo.launch()