Update app.py

app.py

@@ -11,6 +11,7 @@ from docx import Document
 from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
 from io import BytesIO
 import re
+from main import predict_caption, CLIPModel, get_text_embeddings, get_alternative_caption
 
 openai.api_key = "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"
 
@@ -124,11 +125,17 @@ if uploaded_file is not None:
             st.write(radiology_report_with_personal_info)
             st.markdown(download_link(save_as_docx(radiology_report_with_personal_info, "radiology_report.docx"), "radiology_report.docx", "Download Report as DOCX"), unsafe_allow_html=True)
 
-            # Feedback buttons
-            st.header("Thanks for your feedback!")
-            feedback_options = ["Better", "Satisfied", "Worse"]
+            feedback_options = ["Satisfied", "Not Satisfied"]
             selected_feedback = st.radio("Please provide feedback on the generated report:", feedback_options)
-
-            if st.button("Submit Feedback"):
-                st.success("Thanks for providing feedback!")
-                # Implement your feedback handling logic here based on the `selected_feedback` value
+            
+            if selected_feedback == "Not Satisfied":
+                if st.button("Regenerate Report...?"):
+                    with st.spinner("Regenerating report..."):
+                        alternative_caption = get_alternative_caption(image_np, model, text_embeddings, testing_df["caption"])
+                        regenerated_radiology_report = generate_radiology_report(f"Write Complete Radiology Report for this with clinical info, subjective, Assessment, Finding, Impressions, Conclusion and more in proper order : {alternative_caption}")
+            
+                        regenerated_radiology_report_with_personal_info = f"Patient Name: {first_name} {last_name}\nAge: {age}\nGender: {gender}\n\n{regenerated_radiology_report}"
+            
+                        st.header("Regenerated Radiology Report")
+                        st.write(regenerated_radiology_report_with_personal_info)
+                        st.markdown(download_link(save_as_docx(regenerated_radiology_report_with_personal_info, "regenerated_radiology_report.docx"), "regenerated_radiology_report.docx", "Download Regenerated Report as DOCX"), unsafe_allow_html=True)