Update app.py

app.py

@@ -88,8 +88,7 @@ def infer(genre_txt_content, lyrics_txt_content):
     
     # Command and arguments
     command = [
-        "python", "-u",  # Add -u for unbuffered output
-        "infer.py",
+        "python", "infer.py",
         "--stage1_model", "m-a-p/YuE-s1-7B-anneal-en-cot",
         "--stage2_model", "m-a-p/YuE-s2-1B-general",
         "--genre_txt", f"{genre_txt_path}",
@@ -104,47 +103,19 @@ def infer(genre_txt_content, lyrics_txt_content):
     
     # Execute the command
     try:
-        # Run process with real-time output
-        process = subprocess.Popen(
-            command,
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE,
-            text=True,
-            bufsize=1,
-            universal_newlines=True
-        )
-
-        # Print output in real-time
-        while True:
-            output = process.stdout.readline()
-            error = process.stderr.readline()
-            
-            if output:
-                print(f"Output: {output.strip()}")
-            if error:
-                print(f"Error: {error.strip()}")
-                
-            # Check if process has finished
-            if process.poll() is not None:
-                break
+        subprocess.run(command, check=True)
+        print("Command executed successfully!")
         
-        # Get the return code
-        return_code = process.poll()
-        
-        if return_code == 0:
-            print("Command executed successfully!")
-            output_files = os.listdir(output_dir)
-            if output_files:
-                print("Output folder contents:")
-                for file in output_files:
-                    print(f"- {file}")
-            else:
-                print("Output folder is empty.")
+        # Check and print the contents of the output folder
+        output_files = os.listdir(output_dir)
+        if output_files:
+            print("Output folder contents:")
+            for file in output_files:
+                print(f"- {file}")
         else:
-            print(f"Command failed with return code {return_code}")
-            
+            print("Output folder is empty.")
         return None
-    except Exception as e:
+    except subprocess.CalledProcessError as e:
         print(f"Error occurred: {e}")
         return None
     finally: