Update app.py

app.py

@@ -7,28 +7,33 @@ def process_ebook(ebook_file):
     # Ensure input_files folder exists
     input_dir = "input_files"
     os.makedirs(input_dir, exist_ok=True)
-    # Save the uploaded file to the input_files folder
-    ebook_file_path = os.path.join(input_dir, ebook_file.name)
-    
-    # Only copy the file if it's not already in the destination folder
-    if ebook_file_path != ebook_file.name:
-        shutil.copy(ebook_file.name, ebook_file_path)
-    
+
+    # 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:[, {input_files_str} ")
-    
+    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)
+        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:
@@ -43,28 +48,28 @@ def process_ebook(ebook_file):
     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
     output_dir = "output_audiobooks"
     if os.path.exists(output_dir):
-        files = [str(f) for f in os.listdir(output_dir) if os.path.isfile(os.path.join(output_dir, f))]
+        files = [
+            os.path.join(output_dir, f)
+            for f in os.listdir(output_dir)
+            if os.path.isfile(os.path.join(output_dir, f))
+        ]
         return files
     return []
 
-def download_output_files():
-    files = list_output_files()
-    file_links = {file: os.path.join("output_audiobooks", file) for file in files}
-    return file_links
-
 # Gradio Interface
 with gr.Blocks() as gui:
     gr.Markdown("### Ebook to Audiobook Converter")
 
     with gr.Row():
         with gr.Column():
-            ebook_input = gr.File(label="Upload your ebook file (epub, pdf, etc.)")
+            ebook_input = gr.File(
+                label="Upload your ebook file (epub, pdf, etc.)",
+                type='filepath'  # Specify that we want the file path
+            )
             process_button = gr.Button("Start Processing")
             status_output = gr.Textbox(label="Status")
             process_button.click(process_ebook, inputs=ebook_input, outputs=status_output)
@@ -72,8 +77,17 @@ with gr.Blocks() as gui:
         with gr.Column():
             gr.Markdown("### Download Generated Audiobook Files")
             download_button = gr.Button("Reload Files")
-            file_output = gr.File(label="Generated Audiobook Files", interactive=False)
-
-            download_button.click(fn=lambda: download_output_files(), inputs=[], outputs=file_output)
+            file_output = gr.File(
+                label="Generated Audiobook Files",
+                interactive=False,
+                type='file'  # Ensure the files are served correctly
+            )
+
+            # Update the file_output component with the list of output files
+            def update_output_files():
+                files = list_output_files()
+                return files
+
+            download_button.click(fn=update_output_files, inputs=[], outputs=file_output)
 
 gui.launch()