Update app.py

app.py

@@ -3,7 +3,8 @@ import streamlit as st
 import requests
 import os
 import re
-
+from weasyprint import HTML
+from io import BytesIO
 
 # Environment Setup
 HF_API_URL = "https://router.huggingface.co/novita/v3/openai/chat/completions"
@@ -61,7 +62,11 @@ def evaluate_prompt(user_prompt):
     else:
         return f"Error: {response.status_code} - {response.text}"
 
-
+# Function to convert HTML to PDF
+def convert_html_to_pdf(html_content):
+    pdf = HTML(string=html_content).write_pdf()
+    return pdf
+    
 # App UI
 st.set_page_config("PromptPolice", page_icon="🚓", layout="centered")
 st.markdown("""
@@ -114,9 +119,10 @@ if st.button(":mag_right: Evaluate Prompt"):
             </div>
         """, unsafe_allow_html=True)
 
-        # Download button
-        st.download_button("Download Evaluation", result, file_name="evaluation.txt")
-
+        # Download button for PDF
+        pdf_data = convert_html_to_pdf(result_html)
+        st.download_button("Download Evaluation as PDF", pdf_data, file_name="evaluation.pdf")
+        
         # Toast + Balloons + Auto-scroll
         st.toast("✅ Evaluation complete!", icon="🤖")
         st.balloons()