import { HfInference } from '@huggingface/inference'

const hf = new HfInference(process.env.HUGGINGFACE_API_KEY)

export async function POST(request: Request) {
  try {
    const { symptoms, medicalHistory } = await request.json()
    
    const prompt = `Berikan rekomendasi obat dan perawatan dalam Bahasa Indonesia berdasarkan informasi pasien berikut:

Gejala: ${symptoms}
Riwayat Medis: ${medicalHistory}

Harap berikan rekomendasi yang mencakup:
1. Analisis gejala
2. Rekomendasi obat-obatan yang sesuai
3. Pertimbangan interaksi obat
4. Saran perawatan tambahan
5. Tindakan pencegahan

Catatan: Berikan semua informasi dalam Bahasa Indonesia dan gunakan istilah medis yang umum digunakan di Indonesia.`

    const response = await hf.textGeneration({
      model: "nvidia/Llama-3.1-Nemotron-70B-Instruct-HF",
      inputs: prompt,
      parameters: {
        max_new_tokens: 1000,
        temperature: 0.7,
        top_p: 0.95,
        repetition_penalty: 1.15
      }
    })

    return Response.json({ 
      recommendation: response.generated_text 
    })
  } catch (error) {
    return Response.json({ 
      error: 'Gagal menghasilkan rekomendasi' 
    }, { 
      status: 500 
    })
  }
}