fix(transcription): Behebt Start- und API-Fehler in der App [2f488f42]

transcription-tool/backend/services/llm_service.py

@@ -0,0 +1,91 @@
+import os
+import requests
+import logging
+import time
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+
+def call_gemini_api(prompt: str, retries: int = 3, timeout: int = 600) -> str:
+    """
+    Calls the Gemini Pro API with a given prompt.
+
+    Args:
+        prompt: The text prompt to send to the API.
+        retries: The number of times to retry on failure.
+        timeout: The request timeout in seconds.
+
+    Returns:
+        The text response from the API or an empty string if the response is malformed.
+    
+    Raises:
+        Exception: If the API call fails after all retries.
+    """
+    api_key = os.getenv("GEMINI_API_KEY")
+    if not api_key:
+        logging.error("GEMINI_API_KEY environment variable not set.")
+        raise ValueError("API key not found.")
+
+    url = f"https://generativelanguage.googleapis.com/v1/models/gemini-1.5-flash:generateContent?key={api_key}"
+    headers = {'Content-Type': 'application/json'}
+    payload = {
+        "contents": [{
+            "parts": [{"text": prompt}]
+        }],
+        "generationConfig": {
+            "temperature": 0.7,
+            "topK": 40,
+            "topP": 0.95,
+            "maxOutputTokens": 8192,
+        }
+    }
+    
+    for attempt in range(retries):
+        try:
+            response = requests.post(url, headers=headers, json=payload, timeout=timeout)
+            response.raise_for_status()
+            
+            result = response.json()
+
+            if 'candidates' in result and result['candidates']:
+                candidate = result['candidates'][0]
+                if 'content' in candidate and 'parts' in candidate['content']:
+                    # Check for safety ratings
+                    if 'safetyRatings' in candidate:
+                        blocked = any(r.get('blocked') for r in candidate['safetyRatings'])
+                        if blocked:
+                            logging.error(f"API call blocked due to safety ratings: {candidate['safetyRatings']}")
+                            # Provide a more specific error or return a specific string
+                            return "[Blocked by Safety Filter]"
+                    return candidate['content']['parts'][0]['text']
+            
+            # Handle cases where the response is valid but doesn't contain expected content
+            if 'promptFeedback' in result and result['promptFeedback'].get('blockReason'):
+                reason = result['promptFeedback']['blockReason']
+                logging.error(f"Prompt was blocked by the API. Reason: {reason}")
+                return f"[Prompt Blocked: {reason}]"
+
+            logging.warning(f"Unexpected API response structure on attempt {attempt+1}: {result}")
+            return ""
+
+        except requests.exceptions.HTTPError as e:
+            if e.response.status_code in [500, 502, 503, 504] and attempt < retries - 1:
+                wait_time = (2 ** attempt) * 2  # Exponential backoff
+                logging.warning(f"Server Error {e.response.status_code}. Retrying in {wait_time}s...")
+                time.sleep(wait_time)
+                continue
+            logging.error(f"HTTP Error calling Gemini API: {e.response.status_code} {e.response.text}")
+            raise
+        except requests.exceptions.RequestException as e:
+            if attempt < retries - 1:
+                wait_time = (2 ** attempt) * 2
+                logging.warning(f"Connection Error: {e}. Retrying in {wait_time}s...")
+                time.sleep(wait_time)
+                continue
+            logging.error(f"Final Connection Error calling Gemini API: {e}")
+            raise
+        except Exception as e:
+            logging.error(f"An unexpected error occurred: {e}", exc_info=True)
+            raise
+
+    return "" # Should not be reached if retries are exhausted