next

api/routers/analytics_chatbot.py

@@ -53,7 +53,8 @@ def get_analytics(request: AnalyticsChatRequest):
 
     if request.image_base64 and not response_to_cache.get('description'):
         description = generate_analytics_description(user_query, request.image_base64)
-        response_to_cache['description'] = description
+        if description is not None:
+            response_to_cache['description'] = description
 
     redis_client.set(cache_key, json.dumps(response_to_cache), ex=3000)
     return response_to_cache

src/genai/analytics_chatbot/handlers/analytics_description.py

@@ -15,19 +15,21 @@ os.environ['GOOGLE_API_KEY']=os.getenv('GOOGLE_API_KEY')
 # '''
 
 def generate_analytics_description(query,image_base64):
-    if image_base64 != 'string':
-        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
+    if image_base64=='string':
+        return None
+    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
+