Update app.py

app.py

@@ -26,74 +26,84 @@ ALLOWED_FILE_TYPES = ['png', 'jpg', 'jpeg']
 # ======================
 
 def initialize_api_client():
-    """Initialize Groq API client with environment variables"""
+    """Initialize Groq API client"""
     load_dotenv()
     api_key = os.getenv("GROQ_API_KEY")
     if not api_key:
-        st.error("API key not found. Please check the .env configuration.")
-        st.stop()
-    try:
-        return Groq(api_key=api_key)
-    except Exception as e:
-        st.error(f"Error initializing API client: {e}")
+        st.error("API key not found. Please verify .env configuration.")
         st.stop()
+    return Groq(api_key=api_key)
+
 
 def encode_image(image_path):
-    """Convert image file to base64"""
+    """Encode an image to base64"""
     try:
         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 resize_image(image):
+    """Resize the image to reduce size before encoding it to base64"""
+    max_width = 800  # Max width in pixels
+    max_height = 800  # Max height in pixels
+    image.thumbnail((max_width, max_height))
+    return image
+
+
+def compress_image(image):
+    """Compress the image to JPEG format with a lower quality to reduce size"""
+    buffer = io.BytesIO()
+    image.save(buffer, format="JPEG", quality=70)  # Reduce quality to 70%
+    return base64.b64encode(buffer.getvalue()).decode('utf-8')
+
+
 def process_image(uploaded_file):
-    """Convert uploaded image file to base64 string"""
+    """Convert uploaded image file to base64 string with compression"""
     try:
         image = Image.open(uploaded_file)
-        buffer = io.BytesIO()
-        image.save(buffer, format=image.format)
-        return base64.b64encode(buffer.getvalue()).decode('utf-8'), image.format
+        image = resize_image(image)  # Resize the image to reduce its size
+        base64_image = compress_image(image)  # Compress the image
+        return base64_image, "jpeg"  # Use 'jpeg' as format
     except Exception as e:
         st.error(f"Error processing image: {e}")
         return None, None
 
+
 def generate_pdf(report_text, logo_b64):
-    """Generate PDF report with nutrition analysis and logo"""
-    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
+    """Generate a PDF report with logo"""
+    buffer = io.BytesIO()
+    doc = SimpleDocTemplate(buffer, pagesize=letter)
+    styles = getSampleStyleSheet()
+    
+    # Decode the base64 logo image
+    logo_data = base64.b64decode(logo_b64)
+    logo_image = Image.open(io.BytesIO(logo_data))
+    
+    # Resize the logo to fit the page width (you can adjust size if necessary)
+    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)
+    
+    # Create a ReportLab Image element to add the logo to the PDF
+    logo = ReportLabImage(io.BytesIO(logo_data), width=logo_width, height=logo_height)
+    
+    # Build the PDF content
+    story = [
+        logo,  # Add the logo at the top of the page
+        Spacer(1, 12),  # Space after the logo
+        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
+
 
 def generate_analysis(uploaded_file, client):
     """Generate nutrition analysis using AI (Groq API)"""
@@ -138,20 +148,15 @@ def generate_analysis(uploaded_file, client):
         st.error(f"Error communicating with API: {e}")
         return None
 
-
 # ======================
 # UI COMPONENTS
 # ======================
 
 def display_main_interface():
-    """Render the primary user interface"""
+    """Render primary application 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
-    
+    # HTML with inline styles to change text colors
     st.markdown(f"""
         <div style="text-align: center;">
             <img src="data:image/png;base64,{logo_b64}" width="100">
@@ -162,30 +167,28 @@ def display_main_interface():
     
     st.markdown("---")
     
-    # Display analysis results if available
     if st.session_state.get('analysis_result'):
+        # Create two columns: one for download and one for clear button
         col1, col2 = st.columns([1, 1])
         
-        # Column for download button
+        # Left column for the Download button
         with col1:
             pdf_report = generate_pdf(st.session_state.analysis_result, logo_b64)
-            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.")
+            st.download_button("📄 Download Nutrition Report", data=pdf_report, file_name="nutrition_report.pdf", mime="application/pdf")
         
-        # Column for clear button
+        # Right column for the Clear button
         with col2:
             if st.button("Clear Analysis 🗑️"):
-                st.session_state.pop('analysis_result', None)
+                st.session_state.pop('analysis_result')
                 st.rerun()
     
     if st.session_state.get('analysis_result'):
         st.markdown("### 🎯 Nutrition Analysis Report")
         st.info(st.session_state.analysis_result)
 
+
 def render_sidebar(client):
-    """Render the sidebar with image upload and analysis functionality"""
+    """Create sidebar UI elements"""
     with st.sidebar:
         st.subheader("Image Upload")
         uploaded_file = st.file_uploader("Upload Food Image", type=ALLOWED_FILE_TYPES)
@@ -195,18 +198,15 @@ def render_sidebar(client):
             if st.button("Analyze Meal 🍽️"):
                 with st.spinner("Analyzing image..."):
                     report = generate_analysis(uploaded_file, client)
-                    if report:
-                        st.session_state.analysis_result = report
-                        st.rerun()
-                    else:
-                        st.error("Failed to generate analysis. Please try again.")
+                    st.session_state.analysis_result = report
+                    st.rerun()
 
 # ======================
 # APPLICATION ENTRYPOINT
 # ======================
 
 def main():
-    """Main application controller"""
+    """Primary application controller"""
     client = initialize_api_client()
     display_main_interface()
     render_sidebar(client)