Update app.py

app.py

@@ -7,9 +7,11 @@ import numpy as np
 from main import predict_caption, CLIPModel, get_text_embeddings
 import openai
 import base64
-from io import BytesIO
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
+import docx
+from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
+from io import BytesIO
 
 # Set up OpenAI API
 openai.api_key = "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"
@@ -90,32 +92,28 @@ if uploaded_file is not None:
     st.image(image, caption="Uploaded Image", use_column_width=True)
     st.write("")
 
-    if st.button("Generate Report"):
-        with st.spinner("Generating Report...hold on"):
+    if st.button("Generate Caption"):
+        with st.spinner("Generating caption..."):
             image_np = np.array(image)
             caption = show_predicted_caption(image_np)
             st.success(f"Caption: {caption}")
 
-            prompt = f"Write Complete Radiology Report for this: {caption}"
-            report = generate_radiology_report(prompt)
-
-            st.markdown("<h2>Generated Radiology Report</h2>", unsafe_allow_html=True)
-            st.markdown(f"<div class='container'>{report}</div>", unsafe_allow_html=True)
-
-            # PDF download
-            buffer = BytesIO()
-            pdf_buffer = create_pdf(report, buffer)
-            b64 = base64.b64encode(pdf_buffer.getvalue()).decode()
-            href = f'<a href="data:application/octet-stream;base64,{b64}" download="Radiology_Report.pdf">Download Radiology Report as PDF</a>'
-            st.markdown(href, unsafe_allow_html=True)
-
-            # Chatbot for consultation
-            st.markdown("<h2>Need help understanding the Radiological Report?</h2>", unsafe_allow_html=True)
-            st.write("Ask your queries here:")
-            user_question = st.text_input("")
-
-            if st.button("Get Help"):
-                chat_prompt = f"The generated radiology report is: {report}. User asks: {user_question}"
-                chat_answer = chatbot_response(chat_prompt)
-                st.markdown("<h3>Answer:</h3>", unsafe_allow_html=True)
-                st.write(chat_answer)
+            # Add the OpenAI API call here and generate the radiology report
+            radiology_report = f"Write Complete Radiology Report for this: {caption}"
+            container = st.beta_container()
+            with container:
+                st.header("Radiology Report")
+                st.write(radiology_report)
+                st.markdown(download_link(save_as_docx(radiology_report, "radiology_report.docx"), "radiology_report.docx", "Download Report as DOCX"), unsafe_allow_html=True)
+            
+            # Add the chatbot functionality
+            st.header("1-to-1 Consultation")
+            st.write("Ask any questions you have about the radiology report:")
+            user_input = st.text_input("Enter your question:")
+            if user_input:
+                if user_input.lower() == "thank you":
+                    st.write("You're welcome! If you have any more questions, feel free to ask.")
+                else:
+                    # Add the OpenAI API call here and generate the answer to the user's question
+                    answer = f"Answer to the user's question based on the generated radiology report: {user_input}"
+                    st.write(answer)