huggingface_api.py

from huggingface_hub import InferenceClient
from dotenv import load_dotenv
import os
import requests

class HuggingFaceAPI:
    def __init__(self):
        load_dotenv()
        self.api_key = os.getenv("HUGGINGFACE_API_KEY")

    def analyze_text(self, text, prompt):
        try:
            client = InferenceClient(api_key=self.api_key)

            messages = [
	            {
                    "role": "system",
                    "content": prompt
                },
                {
                    "role": "user",
                    "content": f"Aufgabe: Analysiere den hochgeladenen Arztbrief: {text}"
                }
            ]

            completion = client.chat.completions.create(
                model="Qwen/QwQ-32B-Preview", 
                messages=messages, 
                max_tokens=500
            )

            return completion.choices[0].message.content
            

        
        except requests.exceptions.RequestException as e:
            return f"Fehler bei der API-Anfrage: {str(e)}"