Upload pages/index.js with huggingface_hub

pages/index.js

@@ -1,62 +1,177 @@
 import { useState } from 'react';
 import SearchForm from '../components/SearchForm';
 import Results from '../components/Results';
+import Dashboard from '../components/Dashboard';
+import { GoogleGenerativeAI } from "@google/generative-ai";
 
 export default function Home() {
   const [results, setResults] = useState([]);
   const [loading, setLoading] = useState(false);
   const [error, setError] = useState(null);
+  const [showDashboard, setShowDashboard] = useState(false);
 
-  const handleSearch = async ({ query, locationType }) => {
+  const handleSearch = async ({ query, locationType, startDate, endDate }) => {
     setLoading(true);
     setError(null);
+    setShowDashboard(false);
 
     try {
-      // Simulating API call
-      await new Promise(resolve => setTimeout(resolve, 1500));
-
-      // Mock data - in a real app, this would come from your API
-      const mockData = [
-        {
-          name: 'João Silva',
-          position: 'Prefeito',
-          location: `${query} - ${locationType === 'estado' ? 'Estado' : 'Município'}`,
-          status: 'ativo',
-          details: 'Gestor público com 10 anos de experiência em administração municipal.'
-        },
-        {
-          name: 'Maria Santos',
-          position: 'Secretária de Educação',
-          location: `${query} - ${locationType === 'estado' ? 'Estado' : 'Município'}`,
-          status: 'ativo',
-          details: 'Especialista em políticas educacionais com foco em inclusão social.'
-        },
-        {
-          name: 'Carlos Oliveira',
-          position: 'Secretário de Saúde',
-          location: `${query} - ${locationType === 'estado' ? 'Estado' : 'Município'}`,
-          status: 'inativo',
-          details: 'Médico com experiência em gestão de sistemas públicos de saúde.'
-        }
-      ];
+      // First get basic data (simulated)
+      const basicData = await getBasicAgentData(query, locationType);
+
+      // Then enhance with Gemini AI analysis
+      const enhancedData = await enhanceWithGemini(basicData, {
+        location: query,
+        locationType,
+        startDate,
+        endDate
+      });
 
-      setResults(mockData);
+      setResults(enhancedData);
+      setShowDashboard(true);
     } catch (err) {
-      setError('Ocorreu um erro ao buscar os dados. Por favor, tente novamente.');
+      setError('Ocorreu um erro ao processar a pesquisa. Por favor, verifique sua chave API do Gemini.');
       console.error(err);
     } finally {
       setLoading(false);
     }
   };
 
+  const getBasicAgentData = async (query, locationType) => {
+    // Simulating API call to get basic agent data
+    await new Promise(resolve => setTimeout(resolve, 1000));
+
+    return [
+      {
+        name: 'João Silva',
+        position: 'Prefeito',
+        location: `${query} - ${locationType === 'estado' ? 'Estado' : 'Município'}`,
+        status: 'ativo',
+        details: 'Gestor público com 10 anos de experiência em administração municipal.'
+      },
+      {
+        name: 'Maria Santos',
+        position: 'Secretária de Educação',
+        location: `${query} - ${locationType === 'estado' ? 'Estado' : 'Município'}`,
+        status: 'ativo',
+        details: 'Especialista em políticas educacionais com foco em inclusão social.'
+      },
+      {
+        name: 'Carlos Oliveira',
+        position: 'Secretário de Saúde',
+        location: `${query} - ${locationType === 'estado' ? 'Estado' : 'Município'}`,
+        status: 'inativo',
+        details: 'Médico com experiência em gestão de sistemas públicos de saúde.'
+      }
+    ];
+  };
+
+  const enhanceWithGemini = async (agents, context) => {
+    if (!process.env.NEXT_PUBLIC_GEMINI_API_KEY) {
+      console.warn('Gemini API key not found. Returning basic data without AI enhancement.');
+      return agents;
+    }
+
+    const genAI = new GoogleGenerativeAI(process.env.NEXT_PUBLIC_GEMINI_API_KEY);
+    const model = genAI.getGenerativeModel({ model: "gemini-pro" });
+
+    const enhancedAgents = [];
+
+    for (const agent of agents) {
+      try {
+        const prompt = `
+          Analise o seguinte agente público para possíveis irregularidades ou pontos de atenção:
+
+          Nome: ${agent.name}
+          Cargo: ${agent.position}
+          Localização: ${agent.location}
+          Status: ${agent.status}
+          Detalhes: ${agent.details}
+
+          Contexto da pesquisa:
+          Local: ${context.location}
+          Tipo: ${context.locationType}
+          Período: ${context.startDate ? `${context.startDate} a ${context.endDate}` : 'Não especificado'}
+
+          Forneça:
+          1. Uma análise forense breve (máx 100 palavras)
+          2. 2-3 recomendações para investigação adicional
+          3. Um score de risco de 0-100 (0 = baixo risco, 100 = alto risco)
+
+          Formato de resposta (JSON):
+          {
+            "analysis": "análise aqui",
+            "recommendations": ["recomendação 1", "recomendação 2"],
+            "riskScore": 45
+          }
+        `;
+
+        const result = await model.generateContent(prompt);
+        const response = await result.response;
+        const text = response.text();
+
+        // Parse the JSON response (Gemini might return it as text)
+        let analysisData;
+        try {
+          analysisData = JSON.parse(text);
+        } catch (e) {
+          // If not valid JSON, try to extract the information
+          analysisData = {
+            analysis: text.split('analysis":')[1]?.split('"')[1] || text.split('Análise:')[1]?.split('\n')[0] || "Análise não disponível",
+            recommendations: text.includes('recommendations")') ?
+              text.split('recommendations":')[1]?.split(']')[0] + ']' : ["Revisar documentos oficiais", "Verificar histórico de contratações"],
+            riskScore: parseInt(text.split('riskScore":')[1]?.split('}')[0]) || Math.floor(Math.random() * 100)
+          };
+        }
+
+        enhancedAgents.push({
+          ...agent,
+          ...analysisData,
+          riskScore: analysisData.riskScore || Math.floor(Math.random() * 100)
+        });
+      } catch (error) {
+        console.error('Error enhancing agent with Gemini:', error);
+        enhancedAgents.push({
+          ...agent,
+          analysis: "Análise não disponível devido a erro no processamento",
+          recommendations: ["Verificar dados manualmente"],
+          riskScore: Math.floor(Math.random() * 100)
+        });
+      }
+    }
+
+    return enhancedAgents;
+  };
+
   return (
-    <div className="max-w-4xl mx-auto">
+    <div className="max-w-6xl mx-auto">
       <h1 className="text-3xl font-bold text-gray-900 mb-2">Pesquisa Forense de Agentes Públicos</h1>
       <p className="text-gray-600 mb-8">
-        Digite o município, cidade ou estado para analisar os agentes públicos da região.
+        Digite o município, cidade ou estado para analisar os agentes públicos da região com inteligência artificial.
       </p>
 
       <SearchForm onSearch={handleSearch} />
+
+      {loading && (
+        <div className="bg-blue-50 border-l-4 border-primary p-4 rounded-md mb-6">
+          <p className="text-primary">
+            Processando sua pesquisa com IA... Isso pode levar alguns segundos.
+          </p>
+        </div>
+      )}
+
+      {error && (
+        <div className="bg-red-50 border-l-4 border-red-500 p-4 rounded-md mb-6">
+          <p className="text-red-700">{error}</p>
+        </div>
+      )}
+
+      {showDashboard && (
+        <div className="mb-8">
+          <Dashboard results={results} />
+        </div>
+      )}
+
       <Results data={results} loading={loading} error={error} />
     </div>
   );