Update app.py

app.py

@@ -32,7 +32,11 @@ def initialize_api_client():
     if not api_key:
         st.error("API key not found. Please check the .env configuration.")
         st.stop()
-    return Groq(api_key=api_key)
+    try:
+        return Groq(api_key=api_key)
+    except Exception as e:
+        st.error(f"Error initializing API client: {e}")
+        st.stop()
 
 def encode_image(image_path):
     """Convert image file to base64"""
@@ -40,6 +44,10 @@ def encode_image(image_path):
         with open(image_path, "rb") as img_file:
             return base64.b64encode(img_file.read()).decode("utf-8")
     except FileNotFoundError:
+        st.error(f"Error: Image file '{image_path}' not found.")
+        return ""
+    except Exception as e:
+        st.error(f"Error encoding image '{image_path}': {e}")
         return ""
 
 def process_image(uploaded_file):
@@ -55,38 +63,43 @@ def process_image(uploaded_file):
 
 def generate_pdf(report_text, logo_b64):
     """Generate PDF report with nutrition analysis and logo"""
-    buffer = io.BytesIO()
-    doc = SimpleDocTemplate(buffer, pagesize=letter)
-    styles = getSampleStyleSheet()
-    
-    # Decode logo image from base64 and resize
-    logo_data = base64.b64decode(logo_b64)
-    logo_image = Image.open(io.BytesIO(logo_data))
-    logo_width, logo_height = logo_image.size
-    logo_aspect = logo_height / logo_width
-    max_logo_width = 150  # Adjust as needed
-    logo_width = min(logo_width, max_logo_width)
-    logo_height = int(logo_width * logo_aspect)
-    
-    logo = ReportLabImage(io.BytesIO(logo_data), width=logo_width, height=logo_height)
-    
-    # Build PDF content
-    story = [
-        logo,
-        Spacer(1, 12),
-        Paragraph("<b>Nutrition Analysis Report</b>", styles['Title']),
-        Spacer(1, 12),
-        Paragraph(report_text.replace('\n', '<br/>'), styles['BodyText'])
-    ]
-    
-    doc.build(story)
-    buffer.seek(0)
-    return buffer
+    try:
+        buffer = io.BytesIO()
+        doc = SimpleDocTemplate(buffer, pagesize=letter)
+        styles = getSampleStyleSheet()
+        
+        # Decode logo image from base64 and resize
+        logo_data = base64.b64decode(logo_b64)
+        logo_image = Image.open(io.BytesIO(logo_data))
+        logo_width, logo_height = logo_image.size
+        logo_aspect = logo_height / logo_width
+        max_logo_width = 150  # Adjust as needed
+        logo_width = min(logo_width, max_logo_width)
+        logo_height = int(logo_width * logo_aspect)
+        
+        logo = ReportLabImage(io.BytesIO(logo_data), width=logo_width, height=logo_height)
+        
+        # Build PDF content
+        story = [
+            logo,
+            Spacer(1, 12),
+            Paragraph("<b>Nutrition Analysis Report</b>", styles['Title']),
+            Spacer(1, 12),
+            Paragraph(report_text.replace('\n', '<br/>'), styles['BodyText'])
+        ]
+        
+        doc.build(story)
+        buffer.seek(0)
+        return buffer
+    except Exception as e:
+        st.error(f"Error generating PDF: {e}")
+        return None
 
 def generate_analysis(uploaded_file, client):
     """Generate nutrition analysis using AI (Groq API)"""
     base64_image, img_format = process_image(uploaded_file)
     if not base64_image:
+        st.error("Failed to process image. Cannot generate analysis.")
         return None
     
     image_url = f"data:image/{img_format.lower()};base64,{base64_image}"
@@ -118,7 +131,7 @@ def generate_analysis(uploaded_file, client):
         )
         return response.choices[0].message.content
     except Exception as e:
-        st.error(f"API communication error: {e}")
+        st.error(f"Error communicating with API: {e}")
         return None
 
 # ======================
@@ -130,6 +143,10 @@ def display_main_interface():
     logo_b64 = encode_image("src/logo.png")
     
     # Display logo and title
+    if not logo_b64:
+        st.error("Logo image not found or failed to encode.")
+        return
+    
     st.markdown(f"""
         <div style="text-align: center;">
             <img src="data:image/png;base64,{logo_b64}" width="100">
@@ -147,12 +164,15 @@ def display_main_interface():
         # Column for download button
         with col1:
             pdf_report = generate_pdf(st.session_state.analysis_result, logo_b64)
-            st.download_button("📄 Download Nutrition Report", data=pdf_report, file_name="nutrition_report.pdf", mime="application/pdf")
+            if pdf_report:
+                st.download_button("📄 Download Nutrition Report", data=pdf_report, file_name="nutrition_report.pdf", mime="application/pdf")
+            else:
+                st.error("Failed to generate PDF report.")
         
         # Column for clear button
         with col2:
             if st.button("Clear Analysis 🗑️"):
-                st.session_state.pop('analysis_result')
+                st.session_state.pop('analysis_result', None)
                 st.rerun()
     
     if st.session_state.get('analysis_result'):
@@ -170,8 +190,11 @@ def render_sidebar(client):
             if st.button("Analyze Meal 🍽️"):
                 with st.spinner("Analyzing image..."):
                     report = generate_analysis(uploaded_file, client)
-                    st.session_state.analysis_result = report
-                    st.rerun()
+                    if report:
+                        st.session_state.analysis_result = report
+                        st.rerun()
+                    else:
+                        st.error("Failed to generate analysis. Please try again.")
 
 # ======================
 # APPLICATION ENTRYPOINT