Update app.py

app.py

@@ -1738,7 +1738,8 @@ def extract_images_from_pdf(pdf_stream: io.BytesIO):
     except Exception as e:
         raise RuntimeError(f"❌ Error in extract_images_from_pdf: {str(e)}")
 
-def similarity_matching(input_json_path: str, project_folder: str) -> str:
+# def similarity_matching(input_json_path: str, project_folder: str) -> str:
+def similarity_matching(sprites_data: str, project_folder: str) -> str:
     logger.info("🔍 Running similarity matching…")
     os.makedirs(project_folder, exist_ok=True)
 
@@ -1753,8 +1754,8 @@ def similarity_matching(input_json_path: str, project_folder: str) -> str:
     # ==============================
     #    READ SPRITE METADATA
     # ==============================
-    with open(input_json_path, 'r') as f:
-        sprites_data = json.load(f)
+    # with open(input_json_path, 'r') as f:
+    #     sprites_data = json.load(f)
 
     sprite_ids, sprite_base64 = [], []
     for sid, sprite in sprites_data.items():
@@ -2212,7 +2213,11 @@ def save_pdf_to_generated_dir(pdf_stream: io.BytesIO, project_id: str) -> str:
         # 3) Copy the PDF
         # {
         # shutil.copy2(pdf_path, target_pdf)
-        shutil.copy2(pdf_stream, target_pdf)
+        if isinstance(pdf_stream, io.BytesIO):
+            with open(target_pdf, "wb") as f:
+                f.write(pdf_stream.getbuffer())
+        else:
+            shutil.copy2(pdf_stream, target_pdf)
         print(f"Copied PDF from {pdf_stream} → {target_pdf}")
         logger.info(f"Copied PDF from {pdf_stream} → {target_pdf}") 
         # }