Update src/agent/mod.rs

src/agent/mod.rs

@@ -7,10 +7,24 @@ pub async fn chat_handler(
     State(state): State<Arc<AppState>>,
     Json(payload): Json<serde_json::Value>
 ) -> Json<serde_json::Value> {
-    let _ = state.tx.send(json!({"type": "log", "data": "Thinking..."}).to_string());
-    
-    // Gemini API Call Logic here using reqwest
-    // Use model ID: gemini-2.0-flash-exp (current flash preview ID)
-    
-    Json(json!({"status": "ok"}))
+    let user_msg = payload["message"].as_str().unwrap_or("");
+    let api_key = std::env::var("GEMINI_API_KEY").expect("Missing GEMINI_API_KEY");
+
+    let client = reqwest::Client::new();
+    let response = client.post(format!("https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-exp:generateContent?key={}", api_key))
+        .json(&json!({
+            "contents": [{ "parts": [{ "text": user_msg }] }]
+        }))
+        .send()
+        .await;
+
+    if let Ok(res) = response {
+        let body: serde_json::Value = res.json().await.unwrap();
+        let ai_text = body["candidates"][0]["content"]["parts"][0]["text"].as_str().unwrap_or("Error");
+        
+        // Send log to UI via SSE
+        let _ = state.tx.send(json!({"type": "log", "data": ai_text}).to_string());
+    }
+
+    Json(json!({"status": "sent"}))
 }