Update app.py

app.py

@@ -7,8 +7,6 @@ import numpy as np
 from main import predict_caption, CLIPModel, get_text_embeddings
 import openai
 import base64
-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
@@ -76,13 +74,20 @@ def chatbot_response(prompt):
     )
     return response.choices[0].text.strip()
 
+# Add this function to your code
+def save_as_docx(text, filename):
+    document = Document()
+    document.add_paragraph(text)
+    with BytesIO() as output:
+        document.save(output)
+        output.seek(0)
+        return output.getvalue()
 
-def create_pdf(caption, buffer):
-    c = canvas.Canvas(buffer, pagesize=letter)
-    c.drawString(50, 750, caption)
-    c.save()
-    buffer.seek(0)
-    return buffer
+# Add this function to your code
+def download_link(content, filename, link_text):
+    b64 = base64.b64encode(content).decode()
+    href = f'<a href="data:application/octet-stream;base64,{b64}" download="{filename}">{link_text}</a>'
+    return href
 
 
 st.title("RadiXGPT: An Evolution of machine doctors towrads Radiology")
@@ -101,8 +106,8 @@ if uploaded_file is not None:
             st.success(f"Caption: {caption}")
 
             # 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()
+            radiology_report = generate_radiology_report(f"Write Complete Radiology Report for this: {caption}")
+            container = st.container()
             with container:
                 st.header("Radiology Report")
                 st.write(radiology_report)
@@ -122,7 +127,7 @@ if uploaded_file is not None:
                     state.chat_history.append("Bot: 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}"
+                    answer = chatbot_response(f"Answer to the user's question based on the generated radiology report: {user_input}")
                     state.chat_history.append(f"Bot: {answer}")
 
             # Display the chat history