Update crew.py

crew.py

@@ -25,7 +25,7 @@ IMAGE_ANALYSIS_MODEL    = "gemini-2.5-flash-preview-04-17"
 AUDIO_ANALYSIS_MODEL    = "gemini-2.5-flash-preview-04-17"
 VIDEO_ANALYSIS_MODEL    = "gemini-2.5-flash-preview-04-17"
 YOUTUBE_ANALYSIS_MODEL  = "gemini-2.5-flash-preview-04-17"
-DOCUMENT_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
+DOCUMENT_ANALYSIS_MODEL = "gemini-2.0-flash-thinking-exp-01-21"
 CODE_GENERATION_MODEL   = "gemini-2.5-flash-preview-04-17"
 CODE_EXECUTION_MODEL    = "gemini-2.5-flash-preview-04-17"
 
@@ -201,9 +201,6 @@ def run_crew(question, file_path):
             if is_ext(file_path, ".docx"):
                 file_data = read_docx(file_path)
                 contents = [f"{question}\n{file_data}"]
-                print("###")
-                print(file_data)
-                print("###")
             elif is_ext(file_path, ".pptx"):
                 file_data = read_pptx(file_path)
                 contents = [f"{question}\n{file_data}"]
@@ -213,7 +210,12 @@ def run_crew(question, file_path):
             
             response = client.models.generate_content(
                 model=DOCUMENT_ANALYSIS_MODEL,
-                contents=contents
+                contents=contents,
+                config=types.GenerateContentConfig(
+                    thinking_config=types.ThinkingConfig(
+                        thinking_budget=8196
+                    )
+                )
             )
           
             return response.text