Update app.py

app.py

@@ -56,14 +56,6 @@ model = CLIPModel().to(device)
 model.load_state_dict(torch.load("weights.pt", map_location=torch.device('cpu')))
 text_embeddings = torch.load('saved_text_embeddings.pt', map_location=device)
 
-# Add this function to reset the session state
-def reset_session_state():
-    st.session_state.caption = None
-    st.session_state.report = None
-    st.session_state.feedback = None
-    st.session_state.regenerated_report = None
-
-# Add this function to reset the session state
 def reset_session_state():
     st.session_state.caption = None
     st.session_state.report = None
@@ -79,8 +71,8 @@ def show_predicted_caption(image, top_k=8):
     matches = predict_caption(
         image, model, text_embeddings, testing_df["caption"]
     )[:top_k]
-    cleaned_matches = [re.sub(r'\s\(ROCO_\d+\)', '', match) for match in matches]  # Add this line to clean the matches
-    return cleaned_matches  # Return the cleaned_matches instead of matches
+    cleaned_matches = [re.sub(r'\s\(ROCO_\d+\)', '', match) for match in matches]
+    return cleaned_matches
 
 def generate_radiology_report(prompt):
     response = openai.Completion.create(
@@ -92,11 +84,9 @@ def generate_radiology_report(prompt):
         temperature=1,
     )
     report = response.choices[0].text.strip()
-    # Remove reference string from the report
     report = re.sub(r'\(ROCO_\d+\)', '', report).strip()
     return report
 
-
 def save_as_docx(text, filename):
     document = Document()
     document.add_paragraph(text)
@@ -124,7 +114,6 @@ if uploaded_file is not None:
     st.image(image, caption="Uploaded Image", use_column_width=True)
     st.write("")
 
-    # Replace the 'if st.button("Generate Caption"):' block with the following code
     if st.button("Generate Caption") or st.session_state.caption:
         if not st.session_state.caption:
             with st.spinner("Generating caption..."):
@@ -133,26 +122,21 @@ if uploaded_file is not None:
     
         st.success(f"Caption: {st.session_state.caption}")
     
-        # Generate the radiology report
         if not st.session_state.report:
             st.session_state.report = generate_radiology_report(f"Write Complete Radiology Report for this with clinical info, subjective, Assessment, Finding, Impressions, Conclusion and more in proper order : {st.session_state.caption}")
     
-        # Add personal information to the radiology report
         radiology_report_with_personal_info = f"Patient Name: {first_name} {last_name}\nAge: {age}\nGender: {gender}\n\n{st.session_state.report}"
     
         st.header("Radiology Report")
         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_options = ["Satisfied", "Not Satisfied"]
-        st.session_state.feedback = st.radio("Please provide feedback on the generated report:", feedback_options, key="feedback")
         feedback_options = ["Satisfied", "Not Satisfied"]
 
         if not 'feedback' in st.session_state:
             st.session_state.feedback = feedback_options[0]
         st.session_state.feedback = st.radio("Please provide feedback on the generated report:", feedback_options, key="feedback")
 
-    
         if st.session_state.feedback == "Not Satisfied":
             if st.button("Regenerate Report") or st.session_state.regenerated_report:
                 if not st.session_state.regenerated_report:
@@ -164,4 +148,4 @@ if uploaded_file is not None:
     
                 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)
+                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)