Update app.py

app.py

@@ -2,8 +2,21 @@ import os
 import gradio as gr
 import tempfile
 from pathlib import Path
+from utils import get_file_name, run_pdf2text, run_paper2slides
+
+# Create a temporary directory for outputs
+OUTPUT_FOLDER = tempfile.mkdtemp(prefix="pdf_slides_")
+
+def paper2slides(pdf_path: str, logo_path: str='logo.npg'):
+    # Ensure the output directory exists
+    os.makedirs(OUTPUT_FOLDER, exist_ok=True)
+    
+    output_path = os.path.join(OUTPUT_FOLDER, "converted_slides.pptx")
+    file_name = get_file_name(full_path=pdf_path)
+    run_pdf2text(paper_pdf_path=pdf_path, save_json_name=file_name + '.json') ## pdf to json file
+    run_paper2slides(paper_json_name=file_name + '.json', model='llama3_70b', logo_path=logo_path, save_file_name=output_path)  ## json file to slides
+    return output_path
 
-# This is a placeholder function that you would replace with your actual conversion logic
 def convert_pdf_to_slides(pdf_path):
     """
     Convert a PDF file to a slide deck.
@@ -16,18 +29,14 @@ def convert_pdf_to_slides(pdf_path):
     """
     print(f"Processing PDF: {pdf_path}")
     
-    # This is where you would implement your actual conversion logic
-    # For this example, we'll just pretend we're creating a PowerPoint file
+    # Ensure the output directory exists
+    os.makedirs(OUTPUT_FOLDER, exist_ok=True)
     
-    # Create a temporary output file with .pptx extension
-    output_dir = os.path.dirname(pdf_path)
+    # Create output filename in the temporary directory
     output_filename = f"{Path(pdf_path).stem}_slides.pptx"
-    output_path = os.path.join(output_dir, output_filename)
+    output_path = os.path.join(OUTPUT_FOLDER, output_filename)
     
-    # In a real implementation, you would convert the PDF to slides here
-    # For demonstration, we'll just create an empty file
-    with open(output_path, "wb") as f:
-        f.write(b"Example slide deck content")
+    output_path = paper2slides(pdf_path, 'logo.png')
     
     print(f"Conversion complete. Slides saved to: {output_path}")
     return output_path
@@ -82,4 +91,8 @@ with gr.Blocks(title="PDF to Slides Converter") as app:
 
 # Launch the app
 if __name__ == "__main__":
-    app.launch()
+    app.launch()
+
+    # Optional: Clean up the temporary directory when the app exits
+    import shutil
+    shutil.rmtree(OUTPUT_FOLDER, ignore_errors=True)