Update app.py

app.py

@@ -18,6 +18,7 @@ import cachetools
 import hashlib
 import google.generativeai as genai
 from dotenv import load_dotenv
+from decimal import Decimal
 
 app = FastAPI()
 
@@ -34,7 +35,7 @@ if not api_key:
     logger.error("GOOGLE_API_KEY not set")
     raise HTTPException(status_code=500, detail="GOOGLE_API_KEY not set")
 genai.configure(api_key=api_key)
-model = genai.GenerativeModel("gemini-2.5-pro")
+model = genai.GenerativeModel("gemini-2.5-flash")
 
 # Set Tesseract path
 pytesseract.pytesseract.tesseract_cmd = "/usr/bin/tesseract"
@@ -116,11 +117,6 @@ async def process_with_gemini(filename: str, raw_text: str):
 - The 'items' list may have multiple entries, each with detailed attributes.
 - If a field is missing or not found, return an empty value (`""` or `0`) and set `accuracy` to `0.0`.
 - Convert any date found in format :  YYYY-MM-DD
-- For Unit Price Format all numbers in standard decimal notation without exponents:
-- Correct: 0.0000009, 1500000, 0.00123
-- Incorrect: 9e-7, 1.5e+6, 1.23e-3
-
-This applies to all calculations, measurements, and numeric results in your response.
 
 Raw text:
 {raw_text}
@@ -193,7 +189,7 @@ Output JSON:
         json_start = llm_output.find("{")
         json_end = llm_output.rfind("}") + 1
         json_str = llm_output[json_start:json_end]
-        structured_data = json.loads(json_str)
+        structured_data = json.loads(json_str, parse_float=Decimal)
         structured_data_cache[text_hash] = structured_data
         logger.info(f"Gemini processing for {filename}, took {time.time() - start_time:.2f} seconds, {log_memory_usage()}")
         print("Structured Data", structured_data)