Update app.py

app.py

@@ -114,12 +114,14 @@ def extract_invoice_data(file_data, content_type, json_schema):
                 img.save(img_byte_arr, format="PNG", dpi=(300, 300))
                 base64_encoded = base64.b64encode(img_byte_arr.getvalue()).decode('utf-8')
                 base64_images.append(f"data:{content_type};base64,{base64_encoded}")
+            
+            content_type = "image/png"
 
         except Exception as e:
             logger.error(f"Error converting PDF to image: {e}")
             return {"error": "Failed to process PDF"}, None
 
-    else:
+    elif content_type.startswith("image/"):
         # Handle direct image files
         base64_encoded = base64.b64encode(file_data).decode('utf-8')
         base64_images.append(f"data:{content_type.lower()};base64,{base64_encoded}")
@@ -145,6 +147,9 @@ def extract_invoice_data(file_data, content_type, json_schema):
     except Exception as e:
         logger.error(f"Error in OpenAI processing: {e}")
         return {"error": str(e)}, base64_images
+        
+    else:
+        return {"error": f"Unsupported file type: {content_type}"}
 
 def get_content_type_from_s3(file_key):
     """Fetch the content type (MIME type) of a file stored in S3."""