Update app.py

app.py

@@ -56,6 +56,13 @@ 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
@@ -139,6 +146,12 @@ if uploaded_file is not None:
     
         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: