Update app.py

app.py

@@ -1,43 +1,22 @@
 import gradio as gr
 import os
-import shutil
 import subprocess
 
-
+# Declare the global variable
+gradio_input_file = None
 
 def process_ebook(ebook_file):
-    # Ensure input_files folder exists
-    input_dir = "input_files"
-    os.makedirs(input_dir, exist_ok=True)
-
-    # Check if input_files folder is empty
-    if not os.listdir(input_dir):
-        raise FileNotFoundError("The folder 'input_files' is empty. Please upload a file.")
+    global gradio_input_file  # Use the global variable to store the ebook file path
+    gradio_input_file = ebook_file  # Assign the file path to the global variable
     
-    # Get the base name and extension of the uploaded file
+    # Print the name of the uploaded file
     ebook_file_name = os.path.basename(ebook_file)
-    file_name, file_extension = os.path.splitext(ebook_file_name)
-    
-    # Define the new file name with "_Audiobook"
-    new_file_name = f"{file_name}_Audiobook{file_extension}"
-    new_file_path = os.path.join(input_dir, new_file_name)
-    
-    # Copy the uploaded file to the input_files folder with the new name
-    shutil.copy(ebook_file, new_file_path)
-    
-    # print the name of the new file
-    print (f"File duplicated as: {new_file_name} in 'input_files' folder")
+    print(f"Uploaded file: {ebook_file_name}")
 
-    # Print the list of input files in the terminal
-    print(f"Files in input_files folder:\n{input_files_str}")
-
-    if not os.listdir("input_files"):
-        raise FileNotFoundError("The folder 'input_files' is empty.")
-
-    # Call the Auto_VoxNovel.py script and stream its output live
+    # Call the Auto_VoxNovel.py script and pass the file path as an argument
     try:
         process = subprocess.Popen(
-            ["python3", "Auto_VoxNovel.py"],
+            ["python3", "Auto_VoxNovel.py", gradio_input_file],  # Pass gradio_input_file as an argument
             stdout=subprocess.PIPE,
             stderr=subprocess.STDOUT,
             text=True
@@ -50,58 +29,12 @@ def process_ebook(ebook_file):
         process.wait()
 
         if process.returncode == 0:
-            return f"Audiobook is ready! You can now download your files below.\n\nInput files:\n{input_files_str}"
+            return f"Audiobook is ready! You can now download your files."
         else:
-            return f"Error occurred during audiobook processing.\n\nInput files:\n{input_files_str}"
+            return "Error occurred during audiobook processing."
     except Exception as e:
-        return f"Failed to run audiobook script: {str(e)}\n\nInput files:\n{input_files_str}"
-
-
-
+        return f"Failed to run audiobook script: {str(e)}"
 
-#def process_ebook(ebook_file):
-#    # Ensure input_files folder exists
-#    input_dir = "input_files"
-#    os.makedirs(input_dir, exist_ok=True)
-#
-#    # Get the base name of the uploaded file
-#    ebook_file_name = os.path.basename(ebook_file)
-#    ebook_file_path = os.path.join(input_dir, ebook_file_name)
-#
-#    # Copy the file to the input_files directory if it doesn't already exist there
-#    if not os.path.exists(ebook_file_path):
-#        shutil.copy(ebook_file, ebook_file_path)
-#
-#    # List the contents of the input_files folder
-#    input_files_list = os.listdir(input_dir)
-#    input_files_str = "\n".join(input_files_list)
-#
-#    # Print the list of input files in the terminal
-#    print(f"Files in input_files folder:\n{input_files_str}")
-#    if not os.listdir("input_files"):
-#        raise FileNotFoundError("The folder 'input_files' is empty.")
-#
-#    # Call the Auto_VoxNovel.py script and stream its output live
-#    try:
-#        process = subprocess.Popen(
-#            ["python3", "Auto_VoxNovel.py"],
-#            stdout=subprocess.PIPE,
-#            stderr=subprocess.STDOUT,
-#            text=True
-#        )
-#
-#        # Print output in real-time
-#        for line in process.stdout:
-#            print(line, end='')
-#
-#        process.wait()
-#
-#        if process.returncode == 0:
-#            return f"Audiobook is ready! You can now download your files below.\n\nInput files:\n{input_files_str}"
-#        else:
-#            return f"Error occurred during audiobook processing.\n\nInput files:\n{input_files_str}"
-#    except Exception as e:
-#        return f"Failed to run audiobook script: {str(e)}\n\nInput files:\n{input_files_str}"
 
 def list_output_files():
     # List all files in the output directory for downloading