Update utils.py

utils.py

@@ -1,41 +1,35 @@
-import base64
-import os
-
-def encode_image_to_base64(image_path):
-    """
-    Read an image file and convert it to a base64 string.
-    
-    Args:
-        image_path (str): Path to the image file
-        
-    Returns:
-        tuple: (base64_string, mime_type)
-    """
-    try:
-        # Get the file extension
-        _, ext = os.path.splitext(image_path)
-        ext = ext.lower()
-        
-        # Determine MIME type
-        mime_types = {
-            '.png': 'image/png',
-            '.jpg': 'image/jpeg',
-            '.jpeg': 'image/jpeg'
-        }
-        
-        mime_type = mime_types.get(ext, 'image/png')
-        
-        with open(image_path, 'rb') as image_file:
-            # Read the binary data
-            binary_data = image_file.read()
-            # Encode to base64
-            base64_encoded = base64.b64encode(binary_data)
-            # Convert to string and remove b'' prefix
-            return base64_encoded.decode('utf-8'), mime_type
-    except FileNotFoundError:
-        raise FileNotFoundError(f"Image file not found at: {image_path}")
-    except Exception as e:
-        raise Exception(f"Error encoding image: {str(e)}")
-    
-
-# print(encode_image_to_base64("backiee-81547.jpg"))
+import base64
+
+def encode_image_to_base64(uploaded_file):
+    """
+    Convert a Streamlit uploaded file to a base64 string.
+    
+    Args:
+        uploaded_file (UploadedFile): File object from st.file_uploader()
+    
+    Returns:
+        tuple: (base64_string, mime_type)
+    """
+    try:
+        # Get the file extension
+        file_type = uploaded_file.type
+        
+        # Determine MIME type
+        mime_types = {
+            'image/png': 'image/png',
+            'image/jpeg': 'image/jpeg'
+        }
+        
+        mime_type = mime_types.get(file_type, 'image/png')
+        
+        # Read the binary data directly from the uploaded file
+        binary_data = uploaded_file.getvalue()
+        
+        # Encode to base64
+        base64_encoded = base64.b64encode(binary_data)
+        
+        # Convert to string and remove b'' prefix
+        return base64_encoded.decode('utf-8'), mime_type
+        
+    except Exception as e:
+        raise Exception(f"Error encoding image: {str(e)}")