Update crew.py

crew.py

@@ -200,18 +200,15 @@ def run_crew(question, file_path):
             
             if is_ext(file_path, ".docx"):
                 file_data = read_docx_text(file_path)
-                final_question = f"{question}\n{file_data}"
-                contents = [final_question]
+                contents = [f"{question}\n{file_data}"]
+                print(f"=> File data:\n{file_data}")
             elif is_ext(file_path, ".pptx"):
                 file_data = read_pptx_text(file_path)
-                final_question = f"{question}\n{file_data}"
-                contents = [final_question]
+                contents = [f"{question}\n{file_data}"]
+                print(f"=> File data:\n{file_data}")
             else:
                 file = client.files.upload(file=file_path)
                 contents = [file, question]
-
-            print(f"### Initial question: {question}")
-            print(f"### Final question: {final_question}")
             
             response = client.models.generate_content(
                 model=DOCUMENT_ANALYSIS_MODEL,
@@ -239,15 +236,12 @@ def run_crew(question, file_path):
             client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
 
             file_data = read_file_json(file_path)
-
-            final_question = f"{question}\n{file_data}"
-            
-            print(f"### Initial question: {question}")
-            print(f"### Final question: {final_question}")
             
+            print(f"=> File data:\n{file_data}")
+                    
             response = client.models.generate_content(
                 model=CODE_GENERATION_MODEL,
-                contents=[final_question],
+                contents=[f"{question}\n{file_data}"],
                 config=types.GenerateContentConfig(
                     tools=[types.Tool(code_execution=types.ToolCodeExecution)]
                 ),
@@ -429,12 +423,12 @@ def run_crew(question, file_path):
     if file_path:
         question = f"{question} File path: {file_path}."
 
-    initial_answer = crew.kickoff(inputs={"question": question})
-    final_answer = get_final_answer(FINAL_ANSWER_MODEL, question, str(initial_answer))
+    answer = crew.kickoff(inputs={"question": question})
+    final_answer = get_final_answer(FINAL_ANSWER_MODEL, question, str(answer))
 
-    print(f"### Question: {question}")
-    print(f"### Initial answer: {initial_answer}")
-    print(f"### Final answer: {final_answer}")
+    print(f"=> Question: {question}")
+    print(f"=> Answer: {answer}")
+    print(f"=> Final answer: {final_answer}")
     
     return final_answer