Fix Gemini model to use gemini-pro from config

notebooks/04_agent_intelligence.ipynb

@@ -48,8 +48,8 @@
     "AGENT_DIR = MODELS_DIR.parent / \"agent\"\n",
     "AGENT_DIR.mkdir(exist_ok=True)\n",
     "\n",
-    "print(f\"✓ Configuration loaded\")\n",
-    "print(f\"✓ Agent output: {AGENT_DIR}\")"
+    "print(f\"\u2713 Configuration loaded\")\n",
+    "print(f\"\u2713 Agent output: {AGENT_DIR}\")"
    ]
   },
   {
@@ -103,7 +103,7 @@
     "    def to_dict(self) -> Dict:\n",
     "        return asdict(self)\n",
     "\n",
-    "print(\"✓ Task definitions loaded\")"
+    "print(\"\u2713 Task definitions loaded\")"
    ]
   },
   {
@@ -207,7 +207,7 @@
     "                      reverse=True)\n",
     "\n",
     "scoring_engine = DecisionScoringEngine()\n",
-    "print(\"✓ Decision Scoring Engine initialized\")"
+    "print(\"\u2713 Decision Scoring Engine initialized\")"
    ]
   },
   {
@@ -262,12 +262,12 @@
     "        if self.api_key:\n",
     "            try:\n",
     "                genai.configure(api_key=self.api_key)\n",
-    "                self.model = genai.GenerativeModel('gemini-2.0-flash')\n",
-    "                print(\"  ✓ Gemini API connected\")\n",
+    "                self.model = genai.GenerativeModel(CONFIG.get('gemini_model', 'gemini-pro'))\n",
+    "                print(\"  \u2713 Gemini API connected\")\n",
     "            except Exception as e:\n",
-    "                print(f\"  ⚠ Gemini API error: {e}\")\n",
+    "                print(f\"  \u26a0 Gemini API error: {e}\")\n",
     "        else:\n",
-    "            print(\"  ⚠ No Gemini API key found (will use fallback reasoning)\")\n",
+    "            print(\"  \u26a0 No Gemini API key found (will use fallback reasoning)\")\n",
     "    \n",
     "    def analyze_threat(self, threat_data: Dict) -> AgentDecision:\n",
     "        \"\"\"Analyze threat and generate decision with reasoning\"\"\"\n",
@@ -307,7 +307,7 @@
     "                recommended_follow_up=result.get('recommended_follow_up', [])\n",
     "            )\n",
     "        except Exception as e:\n",
-    "            print(f\"  ⚠ Gemini error: {e}\")\n",
+    "            print(f\"  \u26a0 Gemini error: {e}\")\n",
     "            return self._fallback_analyze(threat_data)\n",
     "    \n",
     "    def _fallback_analyze(self, threat_data: Dict) -> AgentDecision:\n",
@@ -339,7 +339,7 @@
     "        )\n",
     "\n",
     "reasoning_engine = GeminiReasoningEngine()\n",
-    "print(\"✓ Gemini Reasoning Engine initialized\")"
+    "print(\"\u2713 Gemini Reasoning Engine initialized\")"
    ]
   },
   {
@@ -435,7 +435,7 @@
     "        }\n",
     "\n",
     "task_queue = AgentTaskQueue()\n",
-    "print(\"✓ Task Queue Manager initialized\")"
+    "print(\"\u2713 Task Queue Manager initialized\")"
    ]
   },
   {
@@ -560,7 +560,7 @@
     "        }\n",
     "\n",
     "agent = CyberForgeAgent()\n",
-    "print(\"✓ CyberForge Agent initialized\")"
+    "print(\"\u2713 CyberForge Agent initialized\")"
    ]
   },
   {
@@ -634,14 +634,14 @@
     "    'severity_weights': DecisionScoringEngine.SEVERITY_WEIGHTS,\n",
     "    'evidence_weights': DecisionScoringEngine.EVIDENCE_WEIGHTS,\n",
     "    'task_priorities': {p.name: p.value for p in TaskPriority},\n",
-    "    'gemini_model': 'gemini-2.0-flash'\n",
+    "    'gemini_model': 'gemini-pro'\n",
     "}\n",
     "\n",
     "config_path = AGENT_DIR / \"agent_config.json\"\n",
     "with open(config_path, 'w') as f:\n",
     "    json.dump(agent_config, f, indent=2)\n",
     "\n",
-    "print(f\"✓ Agent config saved to: {config_path}\")"
+    "print(f\"\u2713 Agent config saved to: {config_path}\")"
    ]
   },
   {
@@ -718,7 +718,7 @@
     "with open(module_path, 'w') as f:\n",
     "    f.write(agent_module)\n",
     "\n",
-    "print(f\"✓ Agent module saved to: {module_path}\")"
+    "print(f\"\u2713 Agent module saved to: {module_path}\")"
    ]
   },
   {
@@ -741,23 +741,23 @@
     "print(\"=\" * 60)\n",
     "\n",
     "print(f\"\"\"\n",
-    "🤖 Agent Capabilities:\n",
+    "\ud83e\udd16 Agent Capabilities:\n",
     "   - Decision Scoring: Weighted threat assessment\n",
     "   - Gemini Integration: AI-powered reasoning\n",
     "   - Task Queue: Priority-based execution\n",
     "   - Action History: Full audit trail\n",
     "\n",
-    "📊 Test Results:\n",
+    "\ud83d\udcca Test Results:\n",
     "   - Action: {decision.action}\n",
     "   - Confidence: {decision.confidence:.2%}\n",
     "   - Risk Level: {decision.risk_level}\n",
     "\n",
-    "📁 Output Files:\n",
+    "\ud83d\udcc1 Output Files:\n",
     "   - Config: {AGENT_DIR}/agent_config.json\n",
     "   - Module: {AGENT_DIR}/cyberforge_agent.py\n",
     "\n",
     "Next step:\n",
-    "  → 05_model_validation.ipynb\n",
+    "  \u2192 05_model_validation.ipynb\n",
     "\"\"\")\n",
     "print(\"=\" * 60)"
    ]
@@ -770,4 +770,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}