qroq_api.py

import os
from groq import Groq

client = Groq(api_key=os.getenv("GROQ_API_KEY"))

def get_diagnosis(prompt):
    response = client.chat.completions.create(
        model="mixtral-8x7b-32768",
        messages=[{"role": "user", "content": prompt}],
        temperature=0.7,
    )
    return response.choices[0].message.content.strip()