Update app.py

app.py

@@ -82,7 +82,8 @@ def main():
 
     if image is not None:
         # Display the captured image
-        st.image(image, caption="Captured Image", use_column_width=True)
+        with st.expander("Expand/collapse the uploaded image:"):
+            st.image(image, caption="Captured Image", use_column_width=True)
 
         # Convert the image to PIL format and resize
         pil_image = Image.open(image)
@@ -148,14 +149,17 @@ def main():
                 with st.expander("Raw output from Gemini"):
                     st.write(response)
 
+                try:
+                    text_from_response = response["candidates"][0]["content"]["parts"][0][
+                        "text"
+                    ]
+                    with st.spinner("Wait for it..."):
+                        st.write(text_from_response)
+                except:
+                    st.warning("Please check the Gemini API as we do not have response from it.")
+
             # Display the response
             try:
-                text_from_response = response["candidates"][0]["content"]["parts"][0][
-                    "text"
-                ]
-                with st.spinner("Wait for it..."):
-                    st.write(text_from_response)
-
                 st.sidebar.success(
                     "Check the box if you want to see a sample retrieved (we had a template of keys, in practice this depends on the stakeholders) information to download (only use this if this is a document-based task)! 👇"
                 )