Ajout API job search avec Gradio

app.py

@@ -1,7 +1,118 @@
 import gradio as gr
 import requests
+import json
 
 SERPAPI_KEY = "7aa26c214c77639dc2be2a61cb01ba2811fde874a36d3b04a38b9823655f6706"
+MIXTRAL_API_KEY = "VhU4tnowxkqtGOIRoyP7qUlhnXj1kjn5"  # Remplace par ta clé Hugging Face
+MIXTRAL_API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1"
+
+def format_jobs_with_mixtral(jobs_data):
+    """
+    Utilise Mixtral pour reformater les offres d'emploi en format optimal pour le LLM chat.
+    
+    Args:
+        jobs_data (dict): Données des offres d'emploi depuis SerpAPI
+        
+    Returns:
+        dict: Format optimisé pour le LLM chat
+    """
+    try:
+        # Créer un prompt pour Mixtral
+        jobs_json = json.dumps(jobs_data, ensure_ascii=False, indent=2)
+        
+        prompt = f"""Tu es un assistant spécialisé dans la présentation d'offres d'emploi. 
+Voici des données JSON d'offres d'emploi que tu dois reformater pour qu'elles soient parfaitement présentables par un LLM chat.
+
+DONNÉES JSON:
+{jobs_json}
+
+INSTRUCTIONS:
+1. Crée un format JSON-RPC 2.0 optimisé
+2. Résume chaque offre de manière claire et concise
+3. Mets en avant les informations essentielles : titre, entreprise, lieu, salaire, lien
+4. Garde les descriptions courtes mais informatives (max 150 caractères)
+5. Assure-toi que le JSON final fait moins de 25000 caractères
+6. Structure le tout pour que le LLM puisse facilement présenter les résultats à l'utilisateur
+
+Réponds UNIQUEMENT avec le JSON formaté, sans explication."""
+
+        # Appel à l'API Mixtral
+        headers = {
+            "Authorization": f"Bearer {MIXTRAL_API_KEY}",
+            "Content-Type": "application/json"
+        }
+        
+        payload = {
+            "inputs": prompt,
+            "parameters": {
+                "max_new_tokens": 2000,
+                "temperature": 0.1,
+                "return_full_text": False
+            }
+        }
+        
+        response = requests.post(MIXTRAL_API_URL, headers=headers, json=payload, timeout=30)
+        
+        if response.status_code == 200:
+            mixtral_response = response.json()
+            if isinstance(mixtral_response, list) and len(mixtral_response) > 0:
+                formatted_text = mixtral_response[0].get("generated_text", "")
+                
+                # Essayer de parser le JSON retourné par Mixtral
+                try:
+                    # Nettoyer la réponse (enlever les éventuels markdown)
+                    if "```json" in formatted_text:
+                        formatted_text = formatted_text.split("```json")[1].split("```")[0]
+                    elif "```" in formatted_text:
+                        formatted_text = formatted_text.split("```")[1].split("```")[0]
+                    
+                    formatted_json = json.loads(formatted_text.strip())
+                    return formatted_json
+                except json.JSONDecodeError:
+                    # Si le parsing échoue, retourner le format original simplifié
+                    pass
+        
+        # Fallback : format simplifié si Mixtral échoue
+        return create_simplified_format(jobs_data)
+        
+    except Exception as e:
+        # En cas d'erreur avec Mixtral, retourner le format simplifié
+        return create_simplified_format(jobs_data)
+
+def create_simplified_format(jobs_data):
+    """
+    Crée un format simplifié en cas d'échec de Mixtral.
+    """
+    if not jobs_data.get("result", {}).get("jobs", {}).get("results"):
+        return jobs_data
+    
+    jobs = jobs_data["result"]["jobs"]["results"]
+    simplified_jobs = []
+    
+    for job in jobs[:6]:  # Limiter à 6 offres max
+        simplified_job = {
+            "titre": job.get("title", ""),
+            "entreprise": job.get("company", ""),
+            "lieu": job.get("location", ""),
+            "salaire": job.get("salary", "Non spécifié"),
+            "description": job.get("description", "")[:120] + "..." if len(job.get("description", "")) > 120 else job.get("description", ""),
+            "lien": job.get("link", "")
+        }
+        simplified_jobs.append(simplified_job)
+    
+    return {
+        "jsonrpc": "2.0",
+        "result": {
+            "status": "SUCCESS",
+            "message": f"✅ {len(simplified_jobs)} offres d'emploi trouvées",
+            "offres": simplified_jobs,
+            "info": {
+                "recherche": jobs_data.get("result", {}).get("search_info", {}).get("query", ""),
+                "lieu": jobs_data.get("result", {}).get("search_info", {}).get("location", "")
+            }
+        },
+        "id": None
+    }
 
 def search_jobs(job_title="", location="", user_desc=None, salary=None, studies=None, domain=None):
     """
@@ -160,112 +271,82 @@ def search_jobs(job_title="", location="", user_desc=None, salary=None, studies=
                 "jsonrpc": "2.0",
                 "result": {
                     "success": False,
-                    "status": "NO_RESULTS_FOUND",
-                    "message": "Aucune offre d'emploi trouvée pour les critères spécifiés",
-                    "search_metadata": {
-                        "original_query": query,
+                    "status": "NO_RESULTS",
+                    "message": "❌ Aucune offre trouvée",
+                    "search_info": {
+                        "query": query,
                         "location": location,
-                        "search_parameters": {
-                            "job_title": job_title,
-                            "location": location,
-                            "user_desc": user_desc,
-                            "salary": salary,
-                            "studies": studies,
-                            "domain": domain
-                        },
-                        "serpapi_parameters": payload,
-                        "search_radius_km": 50,
-                        "fallback_attempted": search_info.get("fallback_query") is not None,
-                        "fallback_query": search_info.get("fallback_query"),
-                        "fallback_results_count": search_info.get("fallback_results", 0)
+                        "fallback_used": search_info.get("fallback_query") is not None
                     },
                     "jobs": {
-                        "total_count": 0,
-                        "displayed_count": 0,
+                        "total": 0,
                         "results": []
                     },
                     "suggestions": [
-                        "Essayez avec un terme de métier plus général",
-                        "Vérifiez l'orthographe de la localisation",
-                        "Élargissez la zone géographique",
-                        "Utilisez des synonymes pour le métier recherché"
-                    ],
-                    "debug_information": search_info
+                        "Essayez un métier plus général",
+                        "Vérifiez l'orthographe de la ville",
+                        "Élargissez la zone géographique"
+                    ]
                 },
                 "id": None
             }
 
-        # Retourner la liste de jobs avec informations très détaillées
+        # Retourner la liste de jobs avec informations essentielles (optimisé pour éviter la troncature)
         jobs_list = []
         for job in jobs_results:
+            # Limiter la description pour éviter les JSON trop lourds
+            description = job.get("description", "")
+            if description and len(description) > 300:
+                description = description[:300] + "..."
+            
             job_info = {
-                "job_id": job.get("job_id", ""),
                 "title": job.get("title", ""),
-                "company_name": job.get("company_name", ""),
-                "company_url": job.get("company_url", ""),
+                "company": job.get("company_name", ""),
                 "location": job.get("location", ""),
-                "via": job.get("via", ""),
-                "description": job.get("description", ""),
-                "posted_at": job.get("posted_at", ""),
-                "schedule_type": job.get("schedule_type", ""),
-                "salary": job.get("salary", ""),
-                "job_highlights": job.get("job_highlights", {}),
-                "related_links": job.get("related_links", []),
-                "thumbnail": job.get("thumbnail", ""),
-                "extensions": job.get("extensions", []),
-                "detected_extensions": job.get("detected_extensions", {}),
-                "apply_options": job.get("apply_options", []),
+                "salary": job.get("salary", "Non spécifié"),
+                "description": description,
+                "posted": job.get("posted_at", ""),
+                "type": job.get("schedule_type", ""),
+                "source": job.get("via", ""),
                 "link": job.get("link", ""),
-                "serpapi_link": job.get("serpapi_link", "")
+                "highlights": {
+                    "qualifications": job.get("job_highlights", {}).get("Qualifications", [])[:3] if job.get("job_highlights", {}).get("Qualifications") else [],
+                    "responsibilities": job.get("job_highlights", {}).get("Responsibilities", [])[:3] if job.get("job_highlights", {}).get("Responsibilities") else []
+                }
             }
             jobs_list.append(job_info)
 
         total_jobs_found = len(jobs_list)
-        displayed_jobs = jobs_list  # Afficher TOUS les résultats, pas de limite
+        # Limiter à 8 résultats max pour éviter la troncature
+        displayed_jobs = jobs_list[:8]
         
-        # Format JSON-RPC 2.0 complet avec toutes les informations
-        return {
+        # Format JSON-RPC 2.0 brut pour Mixtral
+        raw_result = {
             "jsonrpc": "2.0",
             "result": {
                 "success": True,
-                "status": "JOBS_FOUND_SUCCESSFULLY",
-                "message": f"Recherche réussie : {total_jobs_found} offres d'emploi trouvées",
-                "search_metadata": {
-                    "original_query": query,
-                    "processed_keywords": query.split(),
+                "status": "SUCCESS",
+                "message": f"✅ {total_jobs_found} offres trouvées ({len(displayed_jobs)} affichées)",
+                "search_info": {
+                    "query": query,
                     "location": location,
-                    "search_timestamp": None,  # Peut être ajouté si nécessaire
-                    "search_parameters": {
-                        "job_title": job_title,
-                        "location": location,
-                        "user_desc": user_desc,
-                        "salary": salary,
-                        "studies": studies,
-                        "domain": domain
-                    },
-                    "serpapi_parameters": payload,
-                    "search_radius_km": 50,
-                    "fallback_attempted": search_info.get("fallback_query") is not None,
-                    "fallback_query": search_info.get("fallback_query"),
-                    "fallback_results_count": search_info.get("fallback_results", 0),
-                    "serpapi_response_keys": search_info.get("serpapi_response_keys", [])
+                    "fallback_used": search_info.get("fallback_query") is not None
                 },
                 "jobs": {
-                    "total_count": total_jobs_found,
-                    "displayed_count": len(displayed_jobs),
+                    "total": total_jobs_found,
+                    "displayed": len(displayed_jobs),
                     "results": displayed_jobs
                 },
-                "statistics": {
-                    "jobs_with_salary": len([j for j in displayed_jobs if j.get("salary")]),
-                    "jobs_with_description": len([j for j in displayed_jobs if j.get("description")]),
-                    "jobs_with_company_url": len([j for j in displayed_jobs if j.get("company_url")]),
-                    "unique_companies": len(set([j.get("company_name", "") for j in displayed_jobs if j.get("company_name")])),
-                    "job_sources": list(set([j.get("via", "") for j in displayed_jobs if j.get("via")]))
-                },
-                "debug_information": search_info
+                "stats": {
+                    "with_salary": len([j for j in displayed_jobs if j.get("salary") and j.get("salary") != "Non spécifié"]),
+                    "companies": len(set([j.get("company", "") for j in displayed_jobs if j.get("company")]))
+                }
             },
             "id": None
         }
+        
+        # Passer par Mixtral pour optimiser le format pour le LLM chat
+        return format_jobs_with_mixtral(raw_result)
     
     except Exception as e:
         return {