Update app.py

app.py

@@ -98,6 +98,7 @@ def process_cbz_cbr(files):
     final_output = tempfile.mkdtemp()
 
     for file_path in files:
+        file_path = convert_cbr_to_cbz(file_path)
         if file_path.endswith(".cbz"):
             with zipfile.ZipFile(file_path, 'r') as archive:
                 extract_dir = tempfile.mkdtemp()
@@ -125,6 +126,32 @@ def process_cbz_cbr(files):
 
     return zip_buffer.name
 
+import os
+import zipfile
+import rarfile
+import tempfile
+import shutil
+
+def convert_cbr_to_cbz(cbr_path):
+    temp_dir = tempfile.mkdtemp()
+    try:
+        with rarfile.RarFile(cbr_path) as archive:
+            archive.extractall(temp_dir)
+    except rarfile.RarCannotExec:
+        raise RuntimeError("Cannot extract CBR. 'unrar' tool not found. Install 'unrar' or 'unrar-free'.")
+    
+    cbz_path = cbr_path.replace('.cbr', '.cbz')
+    with zipfile.ZipFile(cbz_path, 'w') as zipf:
+        for root, _, files in os.walk(temp_dir):
+            for file in files:
+                file_path = os.path.join(root, file)
+                arcname = os.path.relpath(file_path, temp_dir)
+                zipf.write(file_path, arcname)
+
+    shutil.rmtree(temp_dir)
+    return cbz_path
+
+
 demo = gr.Interface(
     fn=process_cbz_cbr,
     inputs=gr.File(file_types=[".cbz", ".cbr"], file_count="multiple", label="Upload .cbz or .cbr Comic Files"),