Update app.py

app.py

@@ -21,6 +21,22 @@ st.set_page_config(
 
 ALLOWED_FILE_TYPES = ['png', 'jpg', 'jpeg']
 
+# ======================
+# RERUN HELPER FUNCTION
+# ======================
+def rerun():
+    """
+    Helper function to rerun the app.
+    Tries to use st.experimental_rerun if available; otherwise, does nothing.
+    """
+    if hasattr(st, "experimental_rerun"):
+        st.experimental_rerun()
+    else:
+        # Fallback: you might consider raising an exception to force a rerun.
+        # However, this is not recommended for production. Instead, you might simply
+        # notify the user to manually refresh the page.
+        st.warning("Please refresh the page to update the results.")
+
 # ======================
 # UTILITY FUNCTIONS
 # ======================
@@ -50,7 +66,6 @@ def process_image(uploaded_file):
     try:
         image = Image.open(uploaded_file)
         buffer = io.BytesIO()
-        # Ensure image.format is valid; default to PNG if not available.
         fmt = image.format if image.format else "PNG"
         image.save(buffer, format=fmt)
         return base64.b64encode(buffer.getvalue()).decode('utf-8'), fmt
@@ -72,7 +87,6 @@ def generate_pdf(report_text, logo_b64):
         try:
             logo_data = base64.b64decode(logo_b64)
             logo_image = Image.open(io.BytesIO(logo_data))
-            # Resize the logo (adjust max width as needed)
             logo_width, logo_height = logo_image.size
             logo_aspect = logo_height / logo_width
             max_logo_width = 150
@@ -161,10 +175,8 @@ def display_main_interface(logo_b64):
     st.markdown("---")
     
     if st.session_state.get('analysis_result'):
-        # Create two columns: one for download and one for the clear button
         col1, col2 = st.columns(2)
         
-        # Left column for the Download button
         with col1:
             pdf_report = generate_pdf(st.session_state.analysis_result, logo_b64)
             st.download_button("📄 Download Nutrition Report",
@@ -172,11 +184,10 @@ def display_main_interface(logo_b64):
                                file_name="nutrition_report.pdf",
                                mime="application/pdf")
         
-        # Right column for the Clear button
         with col2:
             if st.button("Clear Analysis 🗑️"):
                 st.session_state.pop('analysis_result')
-                st.experimental_rerun()
+                rerun()
     
     if st.session_state.get('analysis_result'):
         st.markdown("### 🎯 Nutrition Analysis Report")
@@ -201,7 +212,7 @@ def render_sidebar(client, logo_b64):
                     report = generate_analysis(uploaded_file, client)
                     if report:
                         st.session_state.analysis_result = report
-                        st.experimental_rerun()
+                        rerun()
                     else:
                         st.error("Failed to generate analysis.")