add: mdfile to pdf

app.py

@@ -7,6 +7,8 @@ from logic import LLMClient, CodeProcessor
 from batch_code_logic_csv import csv_read_batch_code
 import zipfile
 import io
+import markdown2
+import pdfkit
 
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 llm_obj = LLMClient(client)
@@ -48,6 +50,7 @@ if code_dict:
             st.markdown(markdown_output)
 
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+
         st.download_button(
             label=f"Download {unique_key} Result as Markdown",
             data=markdown_output,
@@ -55,6 +58,20 @@ if code_dict:
             mime="text/markdown"
         )
 
+        html_output = markdown2.markdown(markdown_output)
+        pdf_file_path = f"code_analysis_{unique_key}_{timestamp}.pdf"
+        pdfkit.from_string(html_output, pdf_file_path)
+
+        with open(pdf_file_path, "rb") as pdf_file:
+            pdf_data = pdf_file.read()
+
+        st.download_button(
+            label=f"Download {unique_key} Result as PDF",
+            data=pdf_data,
+            file_name=pdf_file_path,
+            mime="application/pdf"
+        )
+
     if st.sidebar.button("Batch Predict"):
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
         all_markdowns = {}
@@ -72,6 +89,20 @@ if code_dict:
                 mime="text/markdown"
             )
 
+            html_output = markdown2.markdown(markdown_output)
+            pdf_file_path = f"code_analysis_{key}_{timestamp}.pdf"
+            pdfkit.from_string(html_output, pdf_file_path)
+
+            with open(pdf_file_path, "rb") as pdf_file:
+                pdf_data = pdf_file.read()
+
+            st.download_button(
+                label=f"Download {key} Result as PDF",
+                data=pdf_data,
+                file_name=pdf_file_path,
+                mime="application/pdf"
+            )
+
         zip_buffer = io.BytesIO()
         with zipfile.ZipFile(zip_buffer, "w") as zip_file:
             for key, markdown_output in all_markdowns.items():

requirements.txt

@@ -1,4 +1,6 @@
 groq
 streamlit
 pandas
-numpy
+numpy
+pdfkit
+markdown2