Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from fpdf import FPDF
+import os
 
-# Initial LLM-generated result (this would be dynamically generated in a real scenario)
 llm_result = """
 Diagnosis: Pneumonia
 
@@ -12,31 +12,38 @@ Prescription:
 - Follow-up in 7 days if symptoms persist
 """
 
-# Function to generate a PDF
 def save_pdf(content):
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", size=12)
     pdf.multi_cell(0, 10, txt=content)
+
     pdf_output_path = "prescription.pdf"
     pdf.output(pdf_output_path)
-    st.success(f"Prescription saved as '{pdf_output_path}'")
+    
+    # Return the path to download
+    return pdf_output_path
 
 # Streamlit app
 def main():
     st.title("Doctor's Assistance: Review and Edit Prescription")
     
     st.write("## Review the LLM-generated prescription and make edits if necessary.")
-    
-    # Display the LLM-generated result in a text area for editing
     edited_text = st.text_area("Edit Prescription", value=llm_result, height=300)
 
-    # Button to save the prescription as PDF
     if st.button("Save Prescription"):
-        if edited_text.strip():  # Check if there's any content
-            save_pdf(edited_text)
+        if edited_text.strip():
+            pdf_file_path = save_pdf(edited_text)
+            st.success("Prescription saved!")
+            with open(pdf_file_path, "rb") as file:
+                st.download_button(
+                    label="Download Prescription as PDF",
+                    data=file,
+                    file_name="prescription.pdf",
+                    mime="application/pdf"
+                )
         else:
             st.error("Prescription content is empty. Please add details.")
 
 if __name__ == "__main__":
-    main()
+    main()