Update app.py

app.py

@@ -84,8 +84,9 @@ def fetch_file_from_s3(file_key):
 def extract_invoice_data(file_data, content_type, json_schema):
     system_prompt = "You are an expert in document data extraction."
 
-    # Convert file to Base64
+        # Convert file to Base64
     base64_encoded = base64.b64encode(file_data).decode('utf-8')
+    base64dataresp = f"data:{content_type};base64,{base64_encoded}" 
 
     # Determine the correct MIME type for OpenAI
     if content_type.startswith("image/"):
@@ -119,7 +120,7 @@ def extract_invoice_data(file_data, content_type, json_schema):
             temperature=0.5,
             max_tokens=16384
         )
-        base64dataresp = f"data:{mime_type};base64,{base64_encoded}"
+        
         # Clean and parse JSON output
         content = response.choices[0].message.content.strip()
         cleaned_content = content.strip().strip('```json').strip('```')