fix incoming message

app.py

@@ -9,6 +9,7 @@ from pypdf import PdfReader
 import requests
 import classificator
 import evaluator
+import extractor_helper
 
 os.environ['TRANSFORMERS_CACHE'] = '/transformers_cache'
 os.environ['HF_HOME'] = '/transformers_cache'
@@ -91,4 +92,5 @@ async def extract(link: InsertedLink):
 
 @app.post("/eval", response_model=float)
 async def eval(eva: EvaModul):
-    return evaluator.evaluate_interview(competences=eva.competences, transcript=eva.transcript)
+    transcript = extractor_helper.extract_technical(eva.competences, eva.transcript)
+    return evaluator.evaluate_interview(competences=eva.competences, transcript=transcript)

extractor_helper.py

@@ -0,0 +1,41 @@
+
+def extract_technical(competences: list[str], transcripts: list[dict]):
+    new_transcripts = {
+        "behavioral": [],
+        "technical": [],
+    }
+    # logger.info(competences)
+
+    for i in range(len(competences)):
+        # new_transcripts[i]= { "competence": competences[i] }
+
+
+        transcript = transcripts[i]
+        # logger.info(transcript)
+
+        if transcript[-1]["question"].startswith("TECHNICAL:"):
+            new_transcripts["behavioral"].append(transcript[:-1])
+            new_transcripts["technical"].append([transcript[-1]])
+        else:
+            new_transcripts["behavioral"].append(transcript)
+            new_transcripts["technical"].append([])
+    
+    return new_transcripts
+
+def extract_competences_and_responses(competences: list[str], transcripts: list[dict]):
+    responses = []
+
+    for i in range(len(competences)):
+        transcript = transcripts[i]
+
+        response = ""
+        for idx, chat in enumerate(transcript):
+            # logger.info(chat)
+            response += chat["answer"]
+
+            if idx < len(transcript) - 1:
+                response += "\n"
+        
+        responses.append(response)
+    
+    return competences, responses

models.py

@@ -65,4 +65,4 @@ class Evaluation(BaseModel):
 
 class EvaModul(BaseModel):
     competences: list[str]
-    transcript: dict
+    transcript: list[dict]