Create app.py

app.py

@@ -0,0 +1,42 @@
+import streamlit as st
+from fpdf import FPDF
+
+# Initial LLM-generated result (this would be dynamically generated in a real scenario)
+llm_result = """
+Diagnosis: Pneumonia
+
+Prescription:
+- Amoxicillin 500 mg, twice daily for 7 days
+- Paracetamol 500 mg, every 6 hours for fever
+- Rest and hydration
+- 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}'")
+
+# 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)
+        else:
+            st.error("Prescription content is empty. Please add details.")
+
+if __name__ == "__main__":
+    main()