Update app.py

app.py

@@ -7,7 +7,7 @@ import os
 import json
 
 app = FastAPI()
-client = InferenceClient(api_key=os.environ['api_key_test'])
+client = InferenceClient(api_key=os.environ['api_key'])
 timing = 1  
 
 class SpecificationInput(BaseModel):
@@ -33,7 +33,7 @@ def generate_specification(input_data: SpecificationInput):
         try:
             messages = [{"role": "Analyste d'affaires axé sur la compréhension des besoins des utilisateurs, la définition des fonctionnalités et leur alignement avec les objectifs du projet", "content": prompt}]
             response = client.chat.completions.create(
-                model="meta-llama/Llama-3.2-3B-Instruct",
+                model=os.environ['model_name'],
                 messages=messages,
                 temperature=0.7
             )
@@ -203,12 +203,12 @@ def generate_specification(input_data: SpecificationInput):
             
             projet["Description_des_fonctionnalités_attendues"]["Fonctionnalités_principales"][f"Fonctionnalité_{str(idx)}"] = {
                 "Nom" : fonctionnalité,
-                "Description": description.text,
-                "Utilisateurs_concernés": utilisateurs_concernés.text,
-                "Bénéfices_attendus": bénéfices_attendus.text,
-                "Tâches_associées": tâches_associées.text,
-                "Conditions_ou_règles_spécifiques": conditions_ou_règles_spécifiques.text,
-                "Flux_de_travail_global": flux_de_travail_global.text
+                "Description": description,
+                "Utilisateurs_concernés": utilisateurs_concernés,
+                "Bénéfices_attendus": bénéfices_attendus,
+                "Tâches_associées": tâches_associées,
+                "Conditions_ou_règles_spécifiques": conditions_ou_règles_spécifiques,
+                "Flux_de_travail_global": flux_de_travail_global
             }
     
         for idx, fonctionnalité in enumerate(liste_des_fonctionnalités, start=1):
@@ -226,7 +226,7 @@ def generate_specification(input_data: SpecificationInput):
             
             projet["Description_des_fonctionnalités_attendues"]["Fonctionnalité_secondaire"][f"Fonctionnalité_{str(idx)}"] = {
                 "Nom" : fonctionnalité,
-                "Détail": response.text
+                "Détail": response
             }
     
         
@@ -238,6 +238,7 @@ def generate_specification(input_data: SpecificationInput):
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
         Liste des Fonctionnalités: {liste_des_fonctionnalités}
+
         Décrivez comment les différentes fonctionnalités interagissent entre elles. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -249,6 +250,7 @@ def generate_specification(input_data: SpecificationInput):
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
         Liste des Fonctionnalités: {liste_des_fonctionnalités}
+
         Précisez les règles qui régissent l'interaction entre les fonctionnalités. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
             
@@ -261,6 +263,7 @@ def generate_specification(input_data: SpecificationInput):
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
         Liste des Fonctionnalités: {liste_des_fonctionnalités}
+
         Précisez les règles qui régissent l'interaction entre les fonctionnalités. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
 
@@ -272,6 +275,7 @@ def generate_specification(input_data: SpecificationInput):
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
         Liste des Fonctionnalités: {liste_des_fonctionnalités}
+
         Mentionnez les exigences de performance pour chaque fonctionnalité, telles que le temps de réponse attendu ou le nombre de transactions par seconde. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -284,6 +288,7 @@ def generate_specification(input_data: SpecificationInput):
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
         Liste des Fonctionnalités: {liste_des_fonctionnalités}
+
         Définissez comment chaque fonctionnalité sera vérifiée pour s'assurer qu'elle répond aux exigences définies. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -294,6 +299,7 @@ def generate_specification(input_data: SpecificationInput):
         Objectif du Projet : {objectif_projet}
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
+
         Veuillez générer une description détaillée incluant l'architecture globale du projet ainsi qu'une description détaillée des composants matériels et logiciels requis pour le projet. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -304,6 +310,7 @@ def generate_specification(input_data: SpecificationInput):
         Objectif du Projet : {objectif_projet}
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
+
         Veuillez générer une description détaillée incluant les exigences d'intégration et d'interopérabilité pour le projet, ainsi qu'une explication de la manière dont le système va s'intégrer avec d'autres systèmes. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -314,6 +321,7 @@ def generate_specification(input_data: SpecificationInput):
         Objectif du Projet : {objectif_projet}
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
+
         Veuillez générer une description détaillée incluant les exigences de sécurité pour le projet, ainsi qu'une explication des mesures de sécurité à mettre en place pour protéger les données et les accès au système. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -324,6 +332,7 @@ def generate_specification(input_data: SpecificationInput):
         Objectif du Projet : {objectif_projet}
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
+
         Veuillez générer une description détaillée incluant les normes et standards à respecter dans ce projet. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -360,7 +369,7 @@ def generate_specification(input_data: SpecificationInput):
             
             
             projet["Liste_des_fonctionnalitées"][fonctionnalité] = {
-                "Tâches": tâches_associées.text.replace('\n','').replace('```','').replace('json','')
+                "Tâches": tâches_associées.replace('\n','').replace('```','').replace('json','')
             }
     
     else:
@@ -539,12 +548,12 @@ def generate_specification(input_data: SpecificationInput):
             
             projet["Description_des_fonctionnalités_attendues"]["Fonctionnalités_principales"][f"Fonctionnalité_{str(idx)}"] = {
                 "Nom" : fonctionnalité,
-                "Description": description.text,
-                "Utilisateurs_concernés": utilisateurs_concernés.text,
-                "Bénéfices_attendus": bénéfices_attendus.text,
-                "Tâches_associées": tâches_associées.text,
-                "Conditions_ou_règles_spécifiques": conditions_ou_règles_spécifiques.text,
-                "Flux_de_travail_global": flux_de_travail_global.text
+                "Description": description,
+                "Utilisateurs_concernés": utilisateurs_concernés,
+                "Bénéfices_attendus": bénéfices_attendus,
+                "Tâches_associées": tâches_associées,
+                "Conditions_ou_règles_spécifiques": conditions_ou_règles_spécifiques,
+                "Flux_de_travail_global": flux_de_travail_global
             }
     
         for idx, fonctionnalité in enumerate(liste_des_fonctionnalités, start=1):
@@ -562,7 +571,7 @@ def generate_specification(input_data: SpecificationInput):
             
             projet["Description_des_fonctionnalités_attendues"]["Fonctionnalité_secondaire"][f"Fonctionnalité_{str(idx)}"] = {
                 "Nom" : fonctionnalité,
-                "Détail": response.text
+                "Détail": response
             }
     
         
@@ -574,6 +583,7 @@ def generate_specification(input_data: SpecificationInput):
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
         Liste des Fonctionnalités: {liste_des_fonctionnalités}
+
         Décrivez comment les différentes fonctionnalités interagissent entre elles. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -585,6 +595,7 @@ def generate_specification(input_data: SpecificationInput):
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
         Liste des Fonctionnalités: {liste_des_fonctionnalités}
+
         Précisez les règles qui régissent l'interaction entre les fonctionnalités. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
             
@@ -598,6 +609,7 @@ def generate_specification(input_data: SpecificationInput):
         Domaine d'Activité du Client : {domaine_activite}
         Liste des Fonctionnalités: {liste_des_fonctionnalités}
         Exigences Techniques : {exigences_techniques}
+
         Précisez les règles qui régissent l'interaction entre les fonctionnalités. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
 
@@ -610,6 +622,7 @@ def generate_specification(input_data: SpecificationInput):
         Domaine d'Activité du Client : {domaine_activite}
         Liste des Fonctionnalités: {liste_des_fonctionnalités}
         Exigences Techniques : {exigences_techniques}
+
         Mentionnez les exigences de performance pour chaque fonctionnalité, telles que le temps de réponse attendu ou le nombre de transactions par seconde. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -623,6 +636,7 @@ def generate_specification(input_data: SpecificationInput):
         Domaine d'Activité du Client : {domaine_activite}
         Liste des Fonctionnalités: {liste_des_fonctionnalités}
         Exigences Techniques : {exigences_techniques}
+
         Définissez comment chaque fonctionnalité sera vérifiée pour s'assurer qu'elle répond aux exigences définies. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -634,6 +648,7 @@ def generate_specification(input_data: SpecificationInput):
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
         Exigences Techniques : {exigences_techniques}
+
         Veuillez générer une description détaillée incluant l'architecture globale du projet ainsi qu'une description détaillée des composants matériels et logiciels requis pour le projet. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -645,6 +660,7 @@ def generate_specification(input_data: SpecificationInput):
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
         Exigences Techniques : {exigences_techniques}
+
         Veuillez générer une description détaillée incluant les exigences d'intégration et d'interopérabilité pour le projet, ainsi qu'une explication de la manière dont le système va s'intégrer avec d'autres systèmes. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -656,6 +672,7 @@ def generate_specification(input_data: SpecificationInput):
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
         Exigences Techniques : {exigences_techniques}
+
         Veuillez générer une description détaillée incluant les exigences de sécurité pour le projet, ainsi qu'une explication des mesures de sécurité à mettre en place pour protéger les données et les accès au système. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -667,6 +684,7 @@ def generate_specification(input_data: SpecificationInput):
         Description du projet : {description_projet}
         Domaine d'Activité du Client : {domaine_activite}
         Exigences Techniques : {exigences_techniques}
+
         Veuillez générer une description détaillée incluant les normes et standards à respecter dans ce projet. Fournissez une réponse complète et homogène sous la forme d'un paragraphe, en {language}.
         """)
     
@@ -703,7 +721,7 @@ def generate_specification(input_data: SpecificationInput):
             
             
             projet["Liste_des_fonctionnalitées"][fonctionnalité] = {
-                "Tâches": tâches_associées.text.replace('\n','').replace('```','').replace('json','')
+                "Tâches": tâches_associées.replace('\n','').replace('```','').replace('json','')
             }
     
-    return projet
+    return projet