Update genai_service.py

genai_service.py

@@ -2,6 +2,7 @@ import os
 from typing import Any
 
 from google import genai
+from google.genai import types
 from pydantic import BaseModel, Field
 
 
@@ -9,7 +10,6 @@ from pydantic import BaseModel, Field
 # CONFIGURATION
 # ============================================================
 
-# Model dapat diganti melalui environment variable HF Space
 GEMINI_MODEL = os.getenv(
     "GEMINI_MODEL",
     "gemini-2.5-pro"
@@ -23,20 +23,21 @@ GEMINI_MODEL = os.getenv(
 class AIAnalysisResult(BaseModel):
     summary: str = Field(
         description=(
-            "Ringkasan singkat hasil deteksi deepfake audio "
+            "Ringkasan hasil deteksi deepfake audio "
             "dalam Bahasa Indonesia."
         )
     )
 
     recommendation: list[str] = Field(
         description=(
-            "Daftar tindakan lanjutan yang praktis dan relevan."
+            "Daftar rekomendasi tindakan lanjutan."
         )
     )
 
     disclaimer: str = Field(
         description=(
-            "Peringatan bahwa hasil deteksi bukan bukti forensik final."
+            "Peringatan bahwa hasil deteksi bukan "
+            "bukti forensik final."
         )
     )
 
@@ -46,13 +47,6 @@ class AIAnalysisResult(BaseModel):
 # ============================================================
 
 def get_gemini_client() -> genai.Client:
-    """
-    Membuat Gemini client.
-
-    API key dibaca otomatis dari environment variable:
-        GEMINI_API_KEY
-    """
-
     api_key = os.getenv(
         "GEMINI_API_KEY"
     )
@@ -68,17 +62,15 @@ def get_gemini_client() -> genai.Client:
 
 
 # ============================================================
-# GENERATE AI EXPLANATION
+# GENERATE AI ANALYSIS
 # ============================================================
 
 def generate_detection_analysis(
     detection_result: dict[str, Any]
 ) -> dict[str, Any]:
     """
-    Membuat penjelasan hasil deteksi menggunakan Gemini API.
-
+    Gemini hanya menjelaskan hasil model TensorFlow.
     Gemini tidak menentukan label real atau fake.
-    Gemini hanya menjelaskan output model TensorFlow.
     """
 
     client = get_gemini_client()
@@ -101,47 +93,35 @@ def generate_detection_analysis(
 Anda adalah AI Analysis Assistant untuk aplikasi pendeteksi
 deepfake audio.
 
-Buat penjelasan yang mudah dipahami oleh pengguna umum dalam
-Bahasa Indonesia.
+Tugas Anda hanya menjelaskan hasil klasifikasi dari model
+TensorFlow. Jangan mengubah hasil klasifikasi.
+
+Gunakan Bahasa Indonesia yang mudah dipahami dan ringkas.
 
-PENTING:
-- Jangan mengubah hasil klasifikasi dari model TensorFlow.
+Aturan penting:
+- Jangan mengubah label prediction.
 - Jangan menyatakan hasil sebagai bukti forensik final.
 - Jangan membuat klaim berlebihan.
-- Berikan rekomendasi praktis.
-- Gunakan kalimat yang ringkas dan jelas.
-
-Berikut hasil deteksi dari model TensorFlow:
-
-prediction                 = {prediction}
-threshold                  = {threshold}
-total_clips                = {total_clips}
-real_clips                 = {real_clips}
-fake_clips                 = {fake_clips}
-average_probability_real   = {average_probability_real}
-average_probability_fake   = {average_probability_fake}
-
-Buat:
-1. Ringkasan hasil analisis.
-2. Rekomendasi tindakan lanjutan.
-3. Disclaimer bahwa hasil ini merupakan indikasi awal dan
-   bukan bukti forensik final.
+- Berikan rekomendasi tindakan yang praktis.
+- Jelaskan bahwa hasil bersifat indikasi awal.
+
+Hasil deteksi model TensorFlow:
+- prediction: {prediction}
+- threshold: {threshold}
+- total_clips: {total_clips}
+- real_clips: {real_clips}
+- fake_clips: {fake_clips}
+- average_probability_real: {average_probability_real}
+- average_probability_fake: {average_probability_fake}
 """
 
     response = client.models.generate_content(
         model=GEMINI_MODEL,
         contents=prompt,
-        config={
-            "response_format": {
-                "text": {
-                    "mime_type": "application/json",
-                    "schema": (
-                        AIAnalysisResult
-                        .model_json_schema()
-                    )
-                }
-            }
-        }
+        config=types.GenerateContentConfig(
+            response_mime_type="application/json",
+            response_schema=AIAnalysisResult
+        )
     )
 
     if not response.text:
@@ -149,11 +129,8 @@ Buat:
             "Gemini API tidak mengembalikan respons."
         )
 
-    parsed_result = (
-        AIAnalysisResult
-        .model_validate_json(
-            response.text
-        )
+    parsed_result = AIAnalysisResult.model_validate_json(
+        response.text
     )
 
     return parsed_result.model_dump()