Update app.py

app.py

@@ -4,7 +4,6 @@ import requests
 from typing import Dict, Any
 from openai import OpenAI
 import httpx
-import time
 
 app = FastAPI()
 
@@ -16,35 +15,39 @@ MODEL = "mistralai/mixtral-8x7b-instruct-v0.1"
 
 # Vérification de la clé API NVIDIA
 if not NVIDIA_API_KEY:
-    raise ValueError("NVIDIA_API_KEY environment variable is not set.")
+    raise ValueError("NVIDIA_API_KEY environment variable is not set. Please set it with a valid NVIDIA API key.")
 
-# Vérification du jeton Hugging Face
+# Vérification du jeton Hugging Face pour la publication des commentaires
 HF_TOKEN = os.getenv("HF_TOKEN")
 if not HF_TOKEN:
-    raise ValueError("HF_TOKEN environment variable is not set.")
+    raise ValueError("HF_TOKEN environment variable is not set. Please set it with a valid Hugging Face API token.")
 
-# Initialisation du client NVIDIA
+# Initialisation du client NVIDIA avec un client HTTP personnalisé
 try:
     client = OpenAI(
         base_url=NVIDIA_API_URL,
         api_key=NVIDIA_API_KEY,
-        http_client=httpx.Client(proxies=None, timeout=30.0)
+        http_client=httpx.Client()
     )
+    print("Successfully initialized NVIDIA API client")
 except Exception as e:
-    raise ValueError(f"Failed to initialize NVIDIA API client: {str(e)}")
+    print(f"Failed to initialize NVIDIA API client: {str(e)}")
+    raise
 
 @app.get("/")
 async def root(request: Request) -> Dict[str, Any]:
     """
     Gère les requêtes GET et renvoie une réponse JSON simple.
     """
-    return {
+    request_details = {
         "method": request.method,
         "host": request.headers.get("host", "unknown"),
         "user_agent": request.headers.get("user-agent", "unknown"),
         "query_test": request.query_params.get("test", None),
         "message": "Webhook server is running. Use POST to interact with the bot."
     }
+    print(f"GET request received: {request_details}")
+    return request_details
 
 @app.post("/")
 async def webhook(request: Request) -> Dict[str, Any]:
@@ -53,7 +56,9 @@ async def webhook(request: Request) -> Dict[str, Any]:
     """
     try:
         if request.headers.get("X-Webhook-Secret") != os.getenv("WEBHOOK_SECRET"):
-            raise HTTPException(status_code=400, detail="Secret incorrect")
+            error_message = "Secret incorrect"
+            print(error_message)
+            raise HTTPException(status_code=400, detail=error_message)
 
         data = await request.json()
         event = data.get("event", {})
@@ -63,6 +68,8 @@ async def webhook(request: Request) -> Dict[str, Any]:
             and event.get("scope") == "discussion.comment"
             and BOT_USERNAME in data.get("comment", {}).get("content", "")
         ):
+            print(f"Processing comment: {data['comment']['content']}")
+
             # Préparation du prompt pour l'API NVIDIA
             messages = [
                 {
@@ -81,44 +88,50 @@ async def webhook(request: Request) -> Dict[str, Any]:
                     max_tokens=100,
                     stream=True
                 )
+                # Collecter les morceaux de texte généré
                 continuation_text = ""
                 for chunk in completion:
                     if chunk.choices[0].delta.content is not None:
                         continuation_text += chunk.choices[0].delta.content
                 continuation_text = continuation_text.strip()
+                print(f"NVIDIA API response: {continuation_text}")
+
                 if not continuation_text:
                     raise ValueError("No text generated by NVIDIA API")
             except Exception as e:
-                raise HTTPException(status_code=500, detail=f"Inference failed: {str(e)}")
+                error_message = f"Inference failed: {str(e)}"
+                print(error_message)
+                raise HTTPException(status_code=500, detail=error_message)
 
-            # Publication du commentaire avec gestion des nouvelles tentatives
+            # Publication du commentaire
             comment_url = data["discussion"]["url"]["api"] + "/comment"
-            for attempt in range(3):  # 3 tentatives maximum
-                try:
-                    comment_response = requests.post(
-                        comment_url,
-                        headers={
-                            "Authorization": f"Bearer {HF_TOKEN}",
-                            "Content-Type": "application/json",
-                        },
-                        json={"comment": continuation_text},
-                    )
-                    comment_response.raise_for_status()
-                    return {"success": True}
-                except requests.exceptions.HTTPError as e:
-                    if "rate-limited" in str(e):
-                        time.sleep(2 ** attempt)  # Backoff exponentiel
-                        continue
-                    raise HTTPException(status_code=500, detail=f"Failed to post comment: {str(e)}")
-
-            raise HTTPException(status_code=429, detail="Rate limit exceeded after retries")
+            print(f"Posting to comment URL: {comment_url}")
+            comment_response = requests.post(
+                comment_url,
+                headers={
+                    "Authorization": f"Bearer {HF_TOKEN}",
+                    "Content-Type": "application/json",
+                },
+                json={"comment": continuation_text},
+            )
+
+            if comment_response.status_code == 200:
+                print("Comment posted successfully")
+                return {"success": True}
+            else:
+                error_message = f"Failed to post comment: {comment_response.text}"
+                print(error_message)
+                raise HTTPException(status_code=500, detail=error_message)
 
+        print("Comment does not meet criteria, no action taken")
         return {"success": False}
 
     except HTTPException as he:
         raise he
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Unexpected error: {str(e)}")
+        error_message = f"Unexpected error in POST request: {str(e)}"
+        print(error_message)
+        raise HTTPException(status_code=500, detail=error_message)
 
 if __name__ == "__main__":
     import uvicorn