Update app.py

app.py

@@ -36,24 +36,24 @@ def main():
     while True:
         try:
             image = camera_input_live()
-        
-        if image is not None:
-            st.image(image)
-
-            if time.time() - st.session_state['last_captured'] > 5:
-                filename = save_image(image)
-                st.session_state['captured_images'].append(filename)
-                st.session_state['last_captured'] = time.time()
-
-            sidebar_html = "<div style='display:flex;flex-direction:column;'>"
-            for img_file in st.session_state['captured_images']:
-                image_base64 = get_image_base64(img_file)
-                sidebar_html += f"<img src='data:image/png;base64,{image_base64}' style='width:100px;'><br>"
-            sidebar_html += "</div>"
-            st.sidebar.markdown("## Captured Images")
-            st.sidebar.markdown(sidebar_html, unsafe_allow_html=True)
-
-            time.sleep(0.1)  # Add a short delay to reduce CPU usage
+            
+            if image is not None:
+                st.image(image)
+    
+                if time.time() - st.session_state['last_captured'] > 5:
+                    filename = save_image(image)
+                    st.session_state['captured_images'].append(filename)
+                    st.session_state['last_captured'] = time.time()
+    
+                sidebar_html = "<div style='display:flex;flex-direction:column;'>"
+                for img_file in st.session_state['captured_images']:
+                    image_base64 = get_image_base64(img_file)
+                    sidebar_html += f"<img src='data:image/png;base64,{image_base64}' style='width:100px;'><br>"
+                sidebar_html += "</div>"
+                st.sidebar.markdown("## Captured Images")
+                st.sidebar.markdown(sidebar_html, unsafe_allow_html=True)
+    
+                time.sleep(0.1)  # Add a short delay to reduce CPU usage
 
         except:
             st.write('.')