additional error handling in call_gemini

app.py

@@ -4,7 +4,6 @@ from transformers import AutoTokenizer, AutoModel
 import anvil.server
 import pathlib
 import textwrap
-import google.generativeai as genai
 import import_ipynb
 from library import call_gpt, call_gemini
 from background_service import BackgroundTaskService
@@ -20,9 +19,6 @@ MESSAGED={'title':'API Server',
 tokenizer = AutoTokenizer.from_pretrained('allenai/specter')
 encoder = AutoModel.from_pretrained('allenai/specter')
 
-# GOOGLE_API_KEY=os.getenv('GOOGLE_API_KEY')
-# genai.configure(api_key=GOOGLE_API_KEY)
-
 service=BackgroundTaskService(max_tasks=10)
 service.register(call_gpt)
 service.register(call_gemini)
@@ -50,12 +46,6 @@ def poll(task_id):
             return str(result)
     else: return str(result)
 
-# @anvil.server.callable
-# def call_gemini(text):
-#     model = genai.GenerativeModel('gemini-pro')
-#     response = model.generate_content(text)
-#     return response.text
-
 @anvil.server.callable
 def encode_anvil(text):
     inputs = tokenizer(text, padding=True, truncation=True, 

library.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -10,7 +10,9 @@
     "import openai\n",
     "import pathlib\n",
     "import textwrap\n",
-    "import google.generativeai as genai"
+    "import google.generativeai as genai #comment this for local deployment and uncomment dummy def below\n",
+    "# class genai:\n",
+    "#     pass"
    ]
   },
   {
@@ -28,12 +30,14 @@
     "        response = model.generate_content(text)\n",
     "    except Exception as e:\n",
     "        return -1,str(e)\n",
-    "    return 0,response.text"
+    "    if 'text' in response.__dict__ :\n",
+    "        return 0,response.text\n",
+    "    else: return -1,'no generation'"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [