Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import streamlit as st
 from pdf_to_svg import convert_pdf_to_svg
 from svg_editor import apply_transformations
+from svg_to_pdf import convert_svg_to_pdf
 import datetime
 
 # Folder paths
@@ -11,23 +12,23 @@ LOG_FOLDER = "./logs/"
 SVG_OUTPUT = os.path.join(OUTPUT_FOLDER, "processed_diagram.svg")
 PDF_OUTPUT = os.path.join(OUTPUT_FOLDER, "processed_diagram.pdf")
 
-
 # Ensure required folders exist
 os.makedirs(INPUT_FOLDER, exist_ok=True)
 os.makedirs(OUTPUT_FOLDER, exist_ok=True)
 os.makedirs(LOG_FOLDER, exist_ok=True)
 
 # Function to log conversion steps
-def log_conversion(status, input_pdf, output_svg):
+def log_conversion(status, input_pdf, output_svg, output_pdf):
     log_path = os.path.join(LOG_FOLDER, "conversion_log.txt")
     with open(log_path, "a") as log_file:
         log_file.write(f"{datetime.datetime.now()} - Status: {status}\n")
         log_file.write(f"Input PDF: {input_pdf}\n")
-        log_file.write(f"Output SVG: {output_svg}\n\n")
+        log_file.write(f"Output SVG: {output_svg}\n")
+        log_file.write(f"Output PDF: {output_pdf}\n\n")
 
 # Streamlit app interface
 def main():
-    st.title("PDF to Editable SVG Converter")
+    st.title("PDF to Editable SVG and PDF Converter")
 
     # File upload
     uploaded_file = st.file_uploader("Upload your PDF file", type="pdf")
@@ -53,16 +54,20 @@ def main():
                 # Step 2: Apply transformations to the SVG
                 apply_transformations(SVG_OUTPUT, brightness, text_scale, line_scale)
 
-                # Step 3: Log the process
-                log_conversion("Success", input_pdf, SVG_OUTPUT)
+                # Step 3: Convert SVG to PDF
+                convert_svg_to_pdf(SVG_OUTPUT, PDF_OUTPUT)
+
+                # Step 4: Log the process
+                log_conversion("Success", input_pdf, SVG_OUTPUT, PDF_OUTPUT)
 
-                # Provide download link
+                # Provide download links
                 st.success("File processed successfully!")
                 st.download_button("Download SVG", open(SVG_OUTPUT, "rb"), file_name="processed_diagram.svg")
-                st.info("You can now import this SVG into CorelDRAW.")
+                st.download_button("Download PDF", open(PDF_OUTPUT, "rb"), file_name="processed_diagram.pdf")
+                st.info("Both SVG and PDF outputs are ready for download.")
             except Exception as e:
-                log_conversion("Failed", input_pdf, SVG_OUTPUT)
+                log_conversion("Failed", input_pdf, SVG_OUTPUT, PDF_OUTPUT)
                 st.error(f"Error processing file: {e}")
 
 if __name__ == "__main__":
-    main()
+    main()