Update app.py

app.py

@@ -6,46 +6,11 @@ from apng import APNG
 import os
 import subprocess
 
-def main():
-    # Gradio Interface
-    with gr.Blocks() as app:
-        gr.Markdown(
-            """
-            # <div align="center"> Ilaria Converter 💖 </div>
-            File conversion Software by Ilaria, support me on [Ko-Fi!](https://ko-fi.com/ilariaowo)  
-    
-            Need help with AI? [Join AI Hub!](https://discord.gg/aihub)
-            """
-        )
-        
-        with gr.Tab('Audio Conversion'):
-            with gr.Row():
-                audio_input = gr.Audio(type='filepath')
-                with gr.Column():
-                    audio_output_format = gr.Dropdown(["wav", "flac", "ogg", "mp3", "aac", "m4a", "m4r"], label="Audio Output Format")
-                    change_bitrate = gr.Checkbox(label="Change Bitrate?")
-                    audio_bitrate = gr.Dropdown(["128", "256", "320"], label="Audio Bitrate")
-                convert_audio_butt = gr.Button(value='Convert Audio', variant='primary')
-                audio_output = gr.Audio(type='filepath', label="Converted Audio")
-                    
-        convert_audio_butt.click(fn=convert_audio, inputs=[audio_input, audio_output_format, change_bitrate, audio_bitrate], outputs=audio_output)
-            
-        with gr.Tab('Image Conversion'):
-            with gr.Row():
-                image_input = gr.Image(type='filepath', image_mode='RGBA')
-                with gr.Column():
-                    image_output_format = gr.Dropdown(["png", "jpg", "tiff", "bmp", "webp", "gif", "apng"], label="Image Output Format")
-                    change_resolution = gr.Checkbox(label="Change Resolution?")
-                    image_width = gr.Number(label="Image Width")
-                    image_height = gr.Number(label="Image Height")
-                convert_image_butt = gr.Button(value='Convert Image', variant='primary')
-                image_output = gr.Image(type='filepath', label="Converted Image")
-                
-        convert_image_butt.click(fn=convert_image, inputs=[image_input, image_output_format, change_resolution, image_width, image_height], outputs=image_output)
-        
-    app.queue(max_size=1022).launch(share=False)
 
 def convert_audio(audio_file, output_format, change_bitrate, bitrate):
+    if audio_file is None:
+        return "Error: No audio file uploaded."
+
     # Get the base name of the input file
     base_name = os.path.splitext(os.path.basename(audio_file))[0]
 
@@ -61,11 +26,17 @@ def convert_audio(audio_file, output_format, change_bitrate, bitrate):
     command.append(output_audio_file)
 
     # Run the command
-    subprocess.run(command, check=True)
+    try:
+        subprocess.run(command, check=True)
+    except subprocess.CalledProcessError as e:
+        return f"Error in conversion: {str(e)}"
     
     return output_audio_file
 
 def convert_image(input_image, output_format, change_resolution, width, height):
+    if input_image is None:
+        return "Error: No image file uploaded."
+    
     base_name = os.path.splitext(os.path.basename(input_image))[0]
     
     out_file = f'{base_name}.{output_format}'
@@ -77,11 +48,56 @@ def convert_image(input_image, output_format, change_resolution, width, height):
 
     command.append(out_file)
 
-    subprocess.run(command, check=True)
-
-    print(f'Converted {input_image} to {out_file}')
+    try:
+        subprocess.run(command, check=True)
+    except subprocess.CalledProcessError as e:
+        return f"Error in conversion: {str(e)}"
 
     return out_file
 
+
+
+def main():
+    # Gradio Interface
+    with gr.Blocks() as app:
+        gr.Markdown(
+            """
+            # <div align="center"> Ilaria Converter 💖 </div>
+            File conversion Software by Ilaria, support her on [Ko-Fi!](https://ko-fi.com/ilariaowo)  
+    
+            Need help with AI? [Join AI Hub!](https://discord.gg/aihub)
+            """
+        )
+        
+        with gr.Tab('Audio Conversion'):
+            with gr.Row():
+                audio_input = gr.Audio(type='filepath', label="Upload Audio File")
+                with gr.Column():
+                    audio_output_format = gr.Dropdown(["wav", "flac", "ogg", "mp3", "aac", "m4a", "m4r"], label="Audio Output Format", info="Choose the desired output format for the audio file.")
+                    change_bitrate = gr.Checkbox(label="Change Bitrate?")
+                    audio_bitrate = gr.Dropdown(["128", "256", "320"], label="Audio Bitrate", info="Choose the bitrate for the audio file.")
+                convert_audio_butt = gr.Button(value='Convert Audio', variant='primary')
+                audio_output = gr.File(label="Download Converted Audio")
+                    
+        convert_audio_butt.click(fn=convert_audio, inputs=[audio_input, audio_output_format, change_bitrate, audio_bitrate], outputs=audio_output)
+            
+        with gr.Tab('Image Conversion'):
+            with gr.Row():
+                image_input = gr.Image(type='filepath', image_mode='RGBA', label="Upload Image")
+                with gr.Column():
+                    image_output_format = gr.Dropdown(["png", "jpg", "tiff", "bmp", "webp", "gif", "apng"], label="Image Output Format", info="Choose the desired output format for the image file.")
+                    change_resolution = gr.Checkbox(label="Change Resolution?")
+                    image_width = gr.Number(label="Image Width", value=1024, visible=False)
+                    image_height = gr.Number(label="Image Height", value=768, visible=False)
+                    
+                convert_image_butt = gr.Button(value='Convert Image', variant='primary')
+                image_output = gr.File(label="Download Converted Image")
+                
+        convert_image_butt.click(fn=convert_image, inputs=[image_input, image_output_format, change_resolution, image_width, image_height], outputs=image_output)
+        with gr.Tab(('')):
+            gr.Markdown("•••---•••")
+    app.queue(max_size=1022).launch(share=False)
+
+
 # Create the Gradio interface
-main()
+main()