added gemini for description

logs/access.log

@@ -1822,3 +1822,5 @@
 2025-11-14 15:21:29,402 | INFO | access_logger | app.py:20 | Response status: 200
 2025-11-14 15:21:38,172 | INFO | access_logger | app.py:18 | Request: POST http://127.0.0.1:8000/api/analytics-chatbot
 2025-11-14 15:21:39,030 | INFO | access_logger | app.py:20 | Response status: 200
+2025-11-14 15:34:22,638 | INFO | access_logger | app.py:18 | Request: POST http://127.0.0.1:8000/api/analytics-chatbot
+2025-11-14 15:34:30,193 | INFO | access_logger | app.py:20 | Response status: 200

requirements.txt

@@ -24,3 +24,4 @@ pytest==8.4.1
 langchain_google_genai==2.1.9
 mangum
 redis==7.0.0
+google-genai

src/genai/analytics_chatbot/handlers/analytics_description.py

@@ -1,31 +1,31 @@
-
-from openai import OpenAI
+from google import genai
+from google.genai import types
 import os
-from ..utils.prompts import analytics_description_prompt
-client = OpenAI(
-base_url="https://router.huggingface.co/v1",
-api_key=os.environ['HUGGINGFACEHUB_ACCESS_TOKEN'],)
+from dotenv import load_dotenv
+load_dotenv()
+os.environ['GOOGLE_API_KEY']=os.getenv('GOOGLE_API_KEY')
+
+
+def analytics_description_prompt(query):
+    return f'''
+You are provided with the user query and the image. 
+Give clear information to the user for their query only by analyzing the image. Don't give any responses outside of the image's context.
+The user query is: \n{query}\n
+'''
 
 def generate_analytics_description(query,image_base64):
-    completion = client.chat.completions.create(
-        model="zai-org/GLM-4.5V:novita",
-        messages=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "text",
-                        "text": analytics_description_prompt(query)
-                    },
-                    {
-                        "type": "image_url",
-                        "image_url": {
-                            "url": f"data:image/jpeg;base64,{image_base64}"
-                        }
-                    }
-                ]
-            }
-        ],
+    client = genai.Client()
+    response = client.models.generate_content(
+        model='gemini-2.5-flash',
+        contents=[
+        types.Part.from_bytes(
+            data=image_base64,
+            mime_type='image/jpeg',
+        ),
+        analytics_description_prompt(query)
+        ]
     )
+    print('The image description:', response.text)
+    return response.text
+
 
-    return completion.choices[0].message.content