Update main.py

main.py

@@ -1,5 +1,7 @@
 import os
 from fastapi import FastAPI, UploadFile, File, HTTPException, Form
+from pydantic import BaseModel, Field
+from typing import Dict, Optional
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from groq import Groq
@@ -7,7 +9,6 @@ import base64
 
 app = FastAPI(title="BSTP-Cameroun-AI-Engine")
 
-# Configuration CORS pour permettre à l'application Frontend de communiquer librement avec l'API
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -83,7 +84,13 @@ FULL_SYSTEM_PROMPT = SYSTEM_PROMPT + "\n\nOFFICIAL BSTP REFERENCE CONTEXT FROM D
 
 class TextRequest(BaseModel):
     text: str
-
+class BenchmarkRequest(BaseModel):
+    company_name: str
+    sector: str  # ex: "Agro-industrie", "BTP", "Hydrocarbures"
+    sme_scores: Dict[str, float]
+    sector_average_scores: Optional[Dict[str, float]] = None
+    total_companies_in_sector: Optional[int] = None
+    current_rank_in_sector: Optional[int] = None
 @app.get("/")
 def read_root():
     return {
@@ -207,4 +214,57 @@ async def audit_document(file: UploadFile = File(...), document_type: str = Form
         return json.loads(response.choices[0].message.content)
         
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Erreur lors de l'analyse OCR/Vision par Groq : {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Erreur lors de l'analyse OCR/Vision par Groq : {str(e)}")
+@app.post("/api/benchmarking")
+async def get_benchmarking_analysis(request: BenchmarkRequest):
+    if not groq_client:
+        raise HTTPException(status_code=500, detail="Le moteur d'IA Groq n'est pas configuré.")
+    
+    # CAS 1 : On n'a pas encore de données dans l'application (Démarrage à vide)
+    if request.sector_average_scores is None:
+        benchmark_prompt = (
+            f"You are a senior industrial consultant specializing in Sub-Saharan African economies.\n"
+            f"The BSTP Cameroon database is currently in its deployment phase and lacks local aggregate data.\n"
+            f"Analyze this Cameroonian SME based on current international standards (UNIDO, ISO, OHADA) "
+            f"and typical economic data available on the internet for the Central African region:\n\n"
+            f"- Company Name: {request.company_name}\n"
+            f"- Sector: {request.sector}\n"
+            f"- SME Actual Scores (out of 10): {request.sme_scores}\n\n"
+            f"YOUR TASK:\n"
+            f"1. Based on market knowledge, general benchmarks for the '{request.sector}' sector in Cameroon, "
+            f"and UNIDO compliance rules, establish a theoretical baseline for each of their 6 axes.\n"
+            f"2. Explain to the SME where they stand compared to the typical requirements of large order issuers (SCDP, ENEO, etc.).\n"
+            f"3. Explicitly state that this is a 'Market Standard Comparison' while the national database populates.\n\n"
+            f"Reply in French or English depending on the query. Keep it formal and highly professional."
+        )
+    
+    # CAS 2 : On a des vraies données en BDD 
+    else:
+        benchmark_prompt = (
+            f"Perform a professional benchmarking analysis based on actual database statistics:\n"
+            f"- Company Name: {request.company_name}\n"
+            f"- Sector: {request.sector}\n"
+            f"- Current Rank: {request.current_rank_in_sector} out of {request.total_companies_in_sector}\n"
+            f"- SME Scores: {request.sme_scores}\n"
+            f"- Database Averages: {request.sector_average_scores}"
+        )
+
+    try:
+        completion = groq_client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=[
+                {"role": "system", "content": FULL_SYSTEM_PROMPT},
+                {"role": "user", "content": benchmark_prompt}
+            ],
+            temperature=0.3,
+            max_tokens=1500
+        )
+        
+        return {
+            "company_name": request.company_name,
+            "sector": request.sector,
+            "mode": "Market Standards (Web/UNIDO)" if request.sector_average_scores is None else "Database Actuals",
+            "benchmarking_report": completion.choices[0].message.content
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Erreur lors du benchmarking : {str(e)}")