Update main.py

main.py

@@ -1850,7 +1850,7 @@ def _get_price_trend_analysis_for_chat(crop_type=None, location=None):
     Analysis:
     """
     try:
-        response_obj = gemini_client.generate_content(model='gemini-2.0-flash', contents=[{'parts': [{'text': prompt}]}])
+        response_obj = gemini_client.models.generate_content(model='gemini-2.0-flash', contents=[{'parts': [{'text': prompt}]}])
         return response_obj.text.strip(), True
     except Exception as e:
         logger.error(f"Gemini error during price trend analysis: {e}")
@@ -1912,7 +1912,7 @@ Return exactly one of the four category names above, with no extra words or punc
 """
         # 2) Call Gemini for classification, with explicit logging
         try:
-            response_obj_gemini = gemini_client.generate_content(
+            response_obj_gemini = gemini_client.models.generate_content(
                 model='gemini-2.0-flash',
                 contents=[{'parts': [{'text': classify_prompt}]}]
             )