update: tempfile

app.py

@@ -9,9 +9,8 @@ import zipfile
 import io
 import markdown2
 import pdfkit
+import tempfile
 
-os.system("apt-get update")
-os.system("apt-get install wkhtmltopdf")
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 llm_obj = LLMClient(client)
 processor = CodeProcessor(llm_obj)
@@ -53,30 +52,35 @@ if code_dict:
 
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
 
-        st.download_button(
-            label=f"Download {unique_key} Result as Markdown",
-            data=markdown_output,
-            file_name=f"code_analysis_{unique_key}_{timestamp}.md",
-            mime="text/markdown"
-        )
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".md") as tmp_md_file:
+            tmp_md_file.write(markdown_output.encode('utf-8'))
+            tmp_md_file_path = tmp_md_file.name
+
+        with open(tmp_md_file_path, "r") as md_file:
+            st.download_button(
+                label=f"Download {unique_key} Result as Markdown",
+                data=md_file,
+                file_name=f"code_analysis_{unique_key}_{timestamp}.md",
+                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 tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_pdf_file:
+            pdf_file_path = tmp_pdf_file.name
+            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"
-        )
+            st.download_button(
+                label=f"Download {unique_key} Result as PDF",
+                data=pdf_file,
+                file_name=f"code_analysis_{unique_key}_{timestamp}.pdf",
+                mime="application/pdf"
+            )
 
     if st.sidebar.button("Batch Predict"):
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
         all_markdowns = {}
+
         for key, code_text in code_dict.items():
             markdown_output = processor.process_code(code_text, model_choice)
             all_markdowns[key] = markdown_output
@@ -84,26 +88,30 @@ if code_dict:
             with st.expander(f"Analysis for {key}"):
                 st.markdown(markdown_output)
 
-            st.download_button(
-                label=f"Download {key} Result as Markdown",
-                data=markdown_output,
-                file_name=f"code_analysis_{key}_{timestamp}.md",
-                mime="text/markdown"
-            )
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".md") as tmp_md_file:
+                tmp_md_file.write(markdown_output.encode('utf-8'))
+                tmp_md_file_path = tmp_md_file.name
+
+            with open(tmp_md_file_path, "r") as md_file:
+                st.download_button(
+                    label=f"Download {key} Result as Markdown",
+                    data=md_file,
+                    file_name=f"code_analysis_{key}_{timestamp}.md",
+                    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 tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_pdf_file:
+                pdf_file_path = tmp_pdf_file.name
+                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"
-            )
+                st.download_button(
+                    label=f"Download {key} Result as PDF",
+                    data=pdf_file,
+                    file_name=f"code_analysis_{key}_{timestamp}.pdf",
+                    mime="application/pdf"
+                )
 
         zip_buffer = io.BytesIO()
         with zipfile.ZipFile(zip_buffer, "w") as zip_file: