Update app.py

app.py

@@ -101,47 +101,55 @@ def process_new_image(image_key, image, kbvqa):
 def run_inference():
     st.title("Run Inference")
 
-    method = st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0)
-    detection_model = st.selectbox("Choose a model for object detection:", ["yolov5", "detic"], index=0)
-    confidence_level = st.slider("Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=0.2 if detection_model == "yolov5" else 0.4, step=0.1)
-
-    # Check for changes in model or confidence level
-    model_changed = (st.session_state.get('detection_model') != detection_model)
-    confidence_changed = (st.session_state.get('confidence_level') != confidence_level)
-
-    if model_changed or confidence_changed:
-        st.session_state['detection_model'] = detection_model
-        st.session_state['confidence_level'] = confidence_level
-        st.warning("Detection model or confidence level changed. Please reload the model, this will take few seconds :)")
-
-
-    
-    # Initialize session state for the model
+    method = st.selectbox(
+        "Choose a method:",
+        ["Fine-Tuned Model", "In-Context Learning (n-shots)"],
+        index=0
+    )
+
+    detection_model = st.selectbox(
+        "Choose a model for object detection:",
+        ["yolov5", "detic"],
+        index=0
+    )
+
+    default_confidence = 0.2 if detection_model == "yolov5" else 0.4
+    confidence_level = st.slider(
+        "Select minimum detection confidence level",
+        min_value=0.1,
+        max_value=0.9,
+        value=default_confidence,
+        step=0.1
+    )
+
+    if 'model_settings' not in st.session_state:
+        st.session_state['model_settings'] = {'detection_model': None, 'confidence_level': None}
+
+    settings_changed = (st.session_state['model_settings']['detection_model'] != detection_model or
+                        st.session_state['model_settings']['confidence_level'] != confidence_level)
+
+    button_label = "Reload Model" if settings_changed else "Load Model"
 
     if method == "Fine-Tuned Model":
         if 'kbvqa' not in st.session_state:
             st.session_state['kbvqa'] = None
-    
-        # Button to load KBVQA models
-
 
-
-        if st.button('Load Model'):
-            if st.session_state.get('kbvqa') and not model_changed and not confidence_changed:
+        if st.button(button_label):
+            if st.session_state['kbvqa'] is not None and not settings_changed:
                 st.write("Model already loaded.")
             else:
                 st.text("Loading the model will take no more than a few minutes . .")
                 st.session_state['kbvqa'] = prepare_kbvqa_model(detection_model)
                 st.session_state['kbvqa'].detection_confidence = confidence_level
-                st.success("Model loaded with updated settings.")
+                st.session_state['model_settings'] = {'detection_model': detection_model, 'confidence_level': confidence_level}
+                st.write("Model is ready for inference.")
 
-        if st.session_state.get('kbvqa'):
-            st.write("Model is ready for inference.")
+        if st.session_state['kbvqa']:
             image_qa_app(st.session_state['kbvqa'])
 
-    else: 
+    else:
         st.write('Model is not ready for inference yet')
-        # here goes the code for n-shot learning
+
 
             
 # Main function
@@ -149,7 +157,7 @@ def main():
 
         
     st.sidebar.title("Navigation")
-    selection = st.sidebar.radio("Go to", ["Home", "Dataset Analysis", "Finetuning and Evaluation Results", "Run Inference", "Dissertation Report"])
+    selection = st.sidebar.radio("Go to", ["Home", "Dataset Analysis", "Finetuning and Evaluation Results", "Run Inference", "Dissertation Report", "Code"])
     st.sidebar.write("More Pages will follow .. ")
 
     if selection == "Home":
@@ -185,9 +193,15 @@ def main():
     elif selection == "Run Inference":
         run_inference()
 
+    elif selection == "Code":
+        st.title("Staye Tuned")
+        st.write("This is a Place Holder until the contents are uploaded.")
+
     elif selection == "More Pages will follow .. ":
         st.title("Staye Tuned")
         st.write("This is a Place Holder until the contents are uploaded.")
+
+