Update src/text_processing.py

src/text_processing.py

@@ -1,12 +1,12 @@
 import os
 import fitz  
 from docx import Document
-import google.generativeai as genai
+from google import genai
 from dotenv import load_dotenv
 
 load_dotenv()
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
-genai.configure(api_key=GOOGLE_API_KEY)
+client = genai.Client(api_key=GOOGLE_API_KEY)
 
 ####################### - TEXT EXTRACTION - #######################
 def extract_text_from_pdf(pdf_path):
@@ -51,9 +51,11 @@ def split_text_by_semantics(text):
     """
 
     try:
-        model = genai.GenerativeModel("gemini-pro")
-        response = model.generate_content(prompt)
+        response = client.models.generate_content(
+            model="gemini-2.0-flash", contents=prompt
+        )
         result_text = response.text.strip()
+        print(result_text)
 
         chunks = result_text.split("- Phần ")
         chunks = [chunk.strip() for chunk in chunks if chunk]
@@ -96,9 +98,10 @@ def generate_explaination_for_chunks(chunks, analysis_level='basic', writting_st
     """
     
     try:
-        model = genai.GenerativeModel("gemini-pro")
-        response = model.generate_content(overview_prompt)
-        overview_text = response.text.strip()
+        response = client.models.generate_content(
+            model="gemini-2.0-flash", contents=overview_prompt
+        )
+        print(response)
         
         explanations = []
         for idx, chunk in enumerate(chunks, start=1):
@@ -111,7 +114,9 @@ def generate_explaination_for_chunks(chunks, analysis_level='basic', writting_st
             Hãy đảm bảo phần tóm tắt không vượt quá {word_upper_limit} từ và không ít hơn {word_lower_limit}.
             """
             
-            part_response = model.generate_content(part_prompt)
+            part_response = response = client.models.generate_content(
+                    model="gemini-2.0-flash", contents=part_prompt
+                )
             explanations.append(part_response.text.strip())
         
         return explanations