Update code.py

code.py

@@ -505,7 +505,13 @@ def structured_output(texts, api_key):
         # Transform items data to dataframe
         items_df = pd.DataFrame(data.get('items', []))
 
-        for column in ['unit', 'price', 'amount']:
+        # Fill 'unit' column with 1 if it is missing
+        if 'unit' not in items_df.columns:
+            items_df['unit'] = 1
+        else:
+            items_df['unit'] = pd.to_numeric(items_df['unit'], errors='coerce').fillna(1)
+
+        for column in ['price', 'amount']:
             items_df[column] = pd.to_numeric(items_df[column], errors='coerce')
 
         items_df = ensure_category(items_df)