Update app.py

app.py

@@ -693,7 +693,7 @@ def main():
                         
                         # Store caption but don't display it yet
                 
-                                        # Detect with CLIP model if loaded
+                                # Detect with CLIP model if loaded
                 if st.session_state.clip_model_loaded:
                     try:
                         with st.spinner("Analyzing image with CLIP model..."):
@@ -748,8 +748,6 @@ def main():
                                         st.session_state.blip_model
                                     )
                                     st.session_state.gradcam_caption = gradcam_caption
-                                    
-                                    # Store caption but don't display it yet
                             
                             # Save results in session state for LLM analysis
                             st.session_state.current_image = image
@@ -762,4 +760,6 @@ def main():
                     except Exception as e:
                         st.error(f"Error analyzing image: {str(e)}")
                         import traceback
-                        st.error(traceback.format_exc())
+                        st.error(traceback.format_exc())
+                else:
+                    st.warning("⚠️ Please load the CLIP model first to perform initial detection.")