added bugs processor

app.py

@@ -165,8 +165,15 @@ with col2:
 
 with col3:
     if st.session_state.get("status") == "loaded":
-        st.image(st.session_state["invoice"], caption="Generated invoice", use_container_width=True)
-        st.image(st.session_state["similar_image"], caption="Similar accident", width=300)
-        st.write(f"Detailed damage description: {st.session_state['detailed_damage_description']}")
-        st.write(f"Damage description: {st.session_state['damage_description']}")
-        st.markdown(get_similar_invoice(st.session_state["invoice_path"]))
+        if st.session_state.get("invoice"):
+            st.image(st.session_state["invoice"], caption="Generated invoice", use_container_width=True)
+        else:
+            st.write("Invoice could not be displayed. Please regenerate.")
+        if st.session_state.get("similar_image"):
+            st.image(st.session_state["similar_image"], caption="Similar accident", width=300)
+        else:
+            st.write("No similar image available.")
+        st.write(f"Detailed damage description: {st.session_state.get('detailed_damage_description', 'N/A')}")
+        st.write(f"Damage description: {st.session_state.get('damage_description', 'N/A')}")
+        if st.session_state.get("invoice_path"):
+            st.markdown(get_similar_invoice(st.session_state["invoice_path"]))