Update analyzer.py

analyzer.py

@@ -1,44 +1,49 @@
 import requests
 import re
 
+from openai import OpenAI
+
+
 class SecurityAuditAnalyzer:
-    def __init__(self, api_key, model="deepseek/deepseek-chat:free", api_url="https://openrouter.ai/api/v1/chat/completions"):
-        self.api_url = api_url
-        self.api_key = api_key
+    def __init__(self, api_key, model="deepseek/deepseek-chat", api_url="https://openrouter.ai/api/v1"):
+        self.client = OpenAI(
+            base_url=api_url,
+            api_key=api_key
+        )
         self.model = model
-        self.headers = {
-            "Content-Type": "application/json",
-            "Authorization": f"Bearer {self.api_key}",
-            "HTTP-Referer": "https://security-audit-api.com",
+        self.extra_headers = {
+            "HTTP-Referer": "https://adisa.africtivistes.com/",  # À remplacer par votre URL en prod
             "X-Title": "Security Audit API"
         }
 
     def analyze_responses(self, responses):
-        """Analyse les réponses d'un audit et retourne une évaluation structurée."""
-        prompt = self._build_prompt(responses)
-        payload = {
-            "model": self.model,
-            "messages": [
-                {"role": "system", "content": self._get_system_prompt()},
-                {"role": "user", "content": prompt}
-            ],
-            "temperature": 0.5
-        }
+        """Analyse les réponses d'un audit en utilisant le client OpenAI"""
         try:
-            response = requests.post(self.api_url, headers=self.headers, json=payload)
-            response.raise_for_status()
-            result = response.json()
-            analysis = result['choices'][0]['message']['content']
+            completion = self.client.chat.completions.create(
+                extra_headers=self.extra_headers,
+                model=self.model,
+                messages=[
+                    {"role": "system", "content": self._get_system_prompt()},
+                    {"role": "user", "content": self._build_prompt(responses)}
+                ],
+                temperature=0.5
+            )
+
+            analysis = completion.choices[0].message.content
             return self._parse_analysis(analysis)
-        except requests.RequestException as e:
+
+        except Exception as e:
             return {
                 "error": str(e),
                 "status": "error",
                 "evaluation": {"score": 0, "level": "Inconnu"},
-                "strengths": [], "weaknesses": [], "recommendations": [],
-                "summary": "Une erreur s'est produite lors de l'analyse."
+                "strengths": [],
+                "weaknesses": [],
+                "recommendations": [],
+                "summary": "Erreur lors de la communication avec l'API"
             }
 
+    # Gardez les autres méthodes (_get_system_prompt, _build_prompt, etc.) identiques
     def _get_system_prompt(self):
         return """Vous êtes un expert en cybersécurité spécialisé dans l'analyse des audits de sécurité.
 Votre tâche est d'analyser les réponses fournies à un audit de sécurité et de produire une évaluation détaillée.