Update app.py

app.py

@@ -60,7 +60,7 @@ print("✅ Standard OCR model loaded.")
 # --- Phase 1: Text Extraction ---
 def classify_image_with_gemini(image: Image.Image):
     """Uses Gemini to classify an image as a 'document' or 'diagram'."""
-    model = genai.GenerativeModel('gemini-1.5-flash-latest')
+    model = genai.GenerativeModel('gemini-2.5-flash')
     prompt = "Is this image primarily a text document or an engineering/technical diagram? Answer with only 'document' or 'diagram'."
     response = model.generate_content([prompt, image])
     classification = response.text.strip().lower()
@@ -69,7 +69,7 @@ def classify_image_with_gemini(image: Image.Image):
 
 def summarize_diagram_with_gemini(image: Image.Image):
     """Uses Gemini to generate a summary of an engineering diagram."""
-    model = genai.GenerativeModel('gemini-1.5-flash-latest')
+    model = genai.GenerativeModel('gemini-2.5-flash')
     prompt = "You are an engineering assistant for Kochi Metro Rail Limited (KMRL). Describe the contents of this technical diagram or engineering drawing in a concise summary. Identify key components and their apparent purpose."
     response = model.generate_content([prompt, image])
     print("✅ Diagram summary successful.")
@@ -125,7 +125,7 @@ def translate_chunk(chunk):
 # --- Phase 3: Gemini Analysis ---
 def generate_structured_json(text_to_analyze):
     """Generates the detailed JSON analysis."""
-    model = genai.GenerativeModel('gemini-1.5-flash-latest')
+    model = genai.GenerativeModel('gemini-2.5-flash')
     prompt = f"You are an AI assistant for KMRL. Analyze this document and extract key info as JSON: {text_to_analyze}"
     json_schema = {"type": "OBJECT", "properties": {"summary": {"type": "STRING"}, "actions_required": {"type": "ARRAY", "items": {"type": "OBJECT", "properties": {"action": {"type": "STRING"}, "priority": {"type": "STRING", "enum": ["High", "Medium", "Low"]}, "deadline": {"type": "STRING"}, "notes": {"type": "STRING"}}, "required": ["action", "priority", "deadline", "notes"]}}, "departments_to_notify": {"type": "ARRAY", "items": {"type": "STRING"}}, "cross_document_flags": {"type": "ARRAY", "items": {"type": "OBJECT", "properties": {"related_document_type": {"type": "STRING"}, "related_issue": {"type": "STRING"}}, "required": ["related_document_type", "related_issue"]}}}, "required": ["summary", "actions_required", "departments_to_notify", "cross_document_flags"]}
     generation_config = genai.types.GenerationConfig(response_mime_type="application/json", response_schema=json_schema)