Public Access
85 lines
3.8 KiB
Python
85 lines
3.8 KiB
Python
import os
|
|
import json
|
|
from dotenv import load_dotenv
|
|
from fasthtml.common import *
|
|
from openai import OpenAI
|
|
|
|
# Chargement du coffre-fort des clés API
|
|
load_dotenv()
|
|
|
|
# Petit test pour vérifier que Python trouve bien les clés au démarrage
|
|
print("Vérification des API :")
|
|
print("- Groq chargée :", "✅" if os.getenv("GROQ_API_KEY") else "❌")
|
|
print("- Gemini chargée :", "✅" if os.getenv("GEMINI_API_KEY") else "❌")
|
|
|
|
# Initialisation de l'application FastHTML
|
|
app, rt = fast_app()
|
|
|
|
# Configuration du client local (Qwen)
|
|
client = OpenAI(
|
|
base_url="http://192.168.1.20:11434/v1",
|
|
api_key="ollama"
|
|
)
|
|
|
|
# Route principale : l'interface utilisateur
|
|
@rt('/')
|
|
def get():
|
|
return Titled("Aethas38 - Orchestrateur",
|
|
Main(
|
|
H1("Interface de Test - Qwen 2.5 (3B)"),
|
|
Form(
|
|
Input(type="text", name="prompt", placeholder="Posez votre question à Qwen...", style="width: 100%; padding: 10px; margin-bottom: 10px;"),
|
|
Button("Envoyer la requête", type="submit", style="padding: 10px 20px; cursor: pointer;"),
|
|
hx_post="/chat", # Envoie la requête à la route /chat
|
|
hx_target="#result", # Injecte la réponse dans la div "result"
|
|
hx_indicator="#loading" # Affiche un indicateur de chargement
|
|
),
|
|
Div("Traitement en cours par le serveur Aethas38...", id="loading", cls="htmx-indicator", style="color: gray; font-style: italic; margin-top: 10px;"),
|
|
Div(id="result", style="margin-top: 20px; padding: 15px; border: 1px solid #444; border-radius: 5px; background-color: #222; color: white;"),
|
|
cls="container"
|
|
)
|
|
)
|
|
|
|
# Route API : Traitement de la requête par l'IA
|
|
@rt('/chat')
|
|
def post(prompt: str):
|
|
try:
|
|
# 1. Le Prompt Système amélioré pour éviter les pièges
|
|
router_prompt = """Tu es l'Aiguilleur (Routeur) du projet Aethas38.
|
|
Analyse la demande de l'utilisateur et choisis le fournisseur d'IA le plus adapté selon ces règles strictes :
|
|
- "groq" : programmation, code, scripts, mathématiques, logique, informatique.
|
|
- "gemini" : rédaction, recettes de cuisine, culture générale, histoire, résumés, art.
|
|
- "local" : UNIQUEMENT pour les salutations (bonjour, test, coucou) ou les banalités très courtes.
|
|
|
|
Format de réponse obligatoire : un objet JSON strict. AUCUN texte d'accompagnement.
|
|
Exemple 1 : {"intention": "créer un script", "fournisseur": "groq"}
|
|
Exemple 2 : {"intention": "recette de gâteau", "fournisseur": "gemini"}
|
|
"""
|
|
|
|
# 2. On interroge notre Qwen local (la GTX 1650 bosse !)
|
|
response = client.chat.completions.create(
|
|
model="qwen2.5:3b",
|
|
messages=[
|
|
{"role": "system", "content": router_prompt},
|
|
{"role": "user", "content": prompt}
|
|
],
|
|
temperature=0.1 # Température basse pour qu'il soit très précis et formel
|
|
)
|
|
|
|
# 3. On récupère et on lit sa décision brute
|
|
raw_decision = response.choices[0].message.content
|
|
|
|
# 4. On l'affiche sur l'interface de test (temporaire, pour voir la magie opérer)
|
|
return Div(
|
|
P(f"🗣️ Votre question : {prompt}", style="color: #888;"),
|
|
P("🧠 Décision de l'Orchestrateur (Qwen) :", style="color: #00d2ff; font-weight: bold; margin-top: 10px;"),
|
|
Pre(raw_decision, style="background: #111; padding: 10px; border-radius: 5px; color: #0f0;"),
|
|
style="border-left: 4px solid #00d2ff; padding-left: 15px; margin-top: 15px;"
|
|
)
|
|
|
|
except Exception as e:
|
|
return P(f"Erreur de l'orchestrateur : {e}", style="color: red; font-weight: bold;")
|
|
|
|
# Lancement du serveur local
|
|
if __name__ == '__main__':
|
|
serve(port=5001) |