Update src/streamlit_app.py

src/streamlit_app.py

@@ -23,15 +23,15 @@ def process_loan_data(df: pd.DataFrame):
     return ecl_df.reset_index()
 
 def get_gemini_decision(segment, pd_val, lgd_val, ead_val, ecl_val):
-    """Gemini-backed risk decision, single-segment call with robust parsing."""
-    model = genai.GenerativeModel("gemini-2.0-flash-lite")
-
-    system_prompt = (
-        "You are a financial risk advisor. "
-        "Return only JSON. "
-        'Schema: {"action":"increase_interest"|"reduce_disbursement"|"maintain","rationale":"string","confidence":float}'
-    )
-    user_prompt = f"""
+    """Gemini-backed risk decision. Single call, valid for all SDK versions."""
+    model = genai.GenerativeModel("gemini-1.5-pro")
+
+    prompt = f"""
+You are a financial risk advisor.
+Analyze the following loan segment and return ONLY valid JSON.
+Schema:
+{{"action":"increase_interest"|"reduce_disbursement"|"maintain","rationale":"string","confidence":float}}
+
 Segment: {segment}
 PD: {pd_val:.3f}
 LGD: {lgd_val:.3f}
@@ -40,26 +40,29 @@ ECL: {ecl_val:,.0f}
 
 Rules:
 - PD > 0.25 ⇒ increase_interest
-- PD between 0.20–0.25 ⇒ reduce_disbursement
+- 0.20 ≤ PD ≤ 0.25 ⇒ reduce_disbursement
 - PD < 0.15 ⇒ maintain
+
 Respond with one JSON object only.
 """
 
     try:
-        resp = model.generate_content(
-            [{"role": "system", "parts": [system_prompt]},
-             {"role": "user", "parts": [user_prompt]}],
-            generation_config={"temperature": 0.1}
-        )
+        resp = model.generate_content(prompt, generation_config={"temperature": 0.1})
         text = resp.text.strip()
+
+        # clean markdown fences if any
         text = re.sub(r"^```json", "", text)
         text = re.sub(r"^```", "", text)
         text = re.sub(r"```$", "", text)
+
+        # extract first {...} JSON block
         match = re.search(r"\{.*\}", text, re.DOTALL)
         if match:
             text = match.group(0)
+
         data = json.loads(text)
         return data
+
     except Exception as e:
         st.warning(f"⚠️ Gemini output parse failed: {e}")
         st.text_area("Raw Gemini output", value=resp.text if 'resp' in locals() else "No response", height=150)