Update app.py

app.py

@@ -19,10 +19,17 @@ logging.getLogger("numba").setLevel(logging.WARNING)
 
 
 def create_vc_fn(tgt_sr, net_g, vc, if_f0, file_index, file_big_npy):
-    def vc_fn(vc_transpose, vc_f0method, vc_index_ratio):
+    def vc_fn(
+        vc_microphone,
+        vc_transpose,
+        vc_f0method,
+        vc_index_ratio,
+        vc_output1,
+        vc_output2
+    ):
         try:
             # Get the recorded audio from the microphone
-            audio, sr = vc_microphone.record(num_frames=16000)  # Adjust the sample rate if needed
+            audio, sr = vc_microphone()
 
             # Your existing processing logic for audio
             times = [0, 0, 0]
@@ -44,11 +51,13 @@ def create_vc_fn(tgt_sr, net_g, vc, if_f0, file_index, file_big_npy):
             print(
                 f"[{datetime.now().strftime('%Y-%m-%d %H:%M')}]: npy: {times[0]}, f0: {times[1]}s, infer: {times[2]}s"
             )
-            return "Success", (tgt_sr, audio_opt)
-        except:
+            vc_output1.update("Success")
+            vc_output2.update((tgt_sr, audio_opt))
+        except Exception as e:
             info = traceback.format_exc()
             print(info)
-            return info, (None, None)
+            vc_output1.update(str(e))
+            vc_output2.update((None, None))
 
     return vc_fn
 
@@ -115,7 +124,7 @@ if __name__ == '__main__':
                         gr.Markdown(
                             '<div align="center">'
                             f'<div>{title}</div>\n' +
-                            (f'<img style="width:auto;height:300px;" src="file/{cover}">' if cover else "") +
+                            (f'<img style="width:auto;height:300px;" src="file/{cover}">' if os.path.exists(cover) else "") +
                             '</div>'
                         )
                     with gr.Row():
@@ -141,5 +150,5 @@ if __name__ == '__main__':
                             vc_output1 = gr.Textbox(label="Output Message")
                             vc_output2 = gr.Audio(label="Output Audio")
 
-                vc_submit.click(vc_fn, [vc_transpose, vc_f0method, vc_index_ratio], [vc_output1, vc_output2])
+                vc_submit.click(vc_fn, vc_microphone, vc_transpose, vc_f0method, vc_index_ratio, vc_output1, vc_output2)
         app.queue(concurrency_count=1, max_size=20, api_open=args.api).launch(share=args.share)