Handle missing openai and zotero info

app.py

@@ -3,7 +3,7 @@ import os
 from dotenv import load_dotenv
 from pathlib import Path
 from functions import (
-    reset_answer,
+    reset_open_ai,
     fetch_collections,
     select_collection,
     reset_collection,
@@ -15,7 +15,7 @@ from models import (
     Messages,
 )
 
-load_dotenv()
+# load_dotenv()
 
 css_style = Path('./style.css').read_text()
 
@@ -84,7 +84,7 @@ with gr.Blocks(css=css_style) as demo:
 
                 answer = gr.HTML(None, elem_id="answer")
 
-    openai_api_key.change(reset_answer, inputs=[], outputs=[
+    openai_api_key.change(reset_open_ai, inputs=[openai_api_key], outputs=[
                           answer], show_progress=False)
 
     zot_api_key.change(
@@ -108,7 +108,8 @@ with gr.Blocks(css=css_style) as demo:
 
     zot_fetch_col_btn.click(
         fn=fetch_collections,
-        inputs=[zot_library_id, zot_library_type, zot_api_key, messages],
+        inputs=[openai_api_key, zot_library_id,
+                zot_library_type, zot_api_key, messages],
         outputs=[zot, zot_collections, zot_selected_col,
                  zot_fetch_col_btn, zot_msg, messages, msg_board],
         show_progress=False,

functions.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import os
 import re
 import requests
 import tempfile
@@ -9,11 +10,58 @@ from lxml import html
 from models import Icons, Message
 
 
-def reset_answer():
+def is_integer(string):
+    try:
+        int(string)
+    except ValueError:
+        return False
+    else:
+        return True
+
+
+def reset_open_ai(openai_api_key):
+    os.environ['OPENAI_API_KEY'] = openai_api_key.strip()
     return gr.HTML.update(value=None)
 
 
-def fetch_collections(id, type, key, messages):
+def fetch_collections(openai_api_key, id, type, key, messages):
+    if openai_api_key == '':
+        messages.append(
+            Message(Icons.ERR, f"Your Open API key is missing. Check out: <a href='https://platform.openai.com/overview' target='_blank'>https://platform.openai.com/overview</a>."))
+        return (
+            None,
+            [],
+            None,
+            gr.Button.update(visible=True),
+            gr.HTML.update(visible=True),
+            messages,
+            gr.HTML.update(value=str(messages)),
+        )
+    if key == '':
+        messages.append(
+            Message(Icons.ERR, f"Your Zotero API key is missing. Click <a href='https: // www.zotero.org/settings/keys/new' target='_blank'>here</a> to create a new one."))
+        return (
+            None,
+            [],
+            None,
+            gr.Button.update(visible=True),
+            gr.HTML.update(visible=True),
+            messages,
+            gr.HTML.update(value=str(messages)),
+        )
+    if not is_integer(id):
+        messages.append(
+            Message(Icons.ERR, f"Your Zotero ID should be an integer."))
+        return (
+            None,
+            [],
+            None,
+            gr.Button.update(visible=True),
+            gr.HTML.update(visible=True),
+            messages,
+            gr.HTML.update(value=str(messages)),
+        )
+
     zot = zotero.Zotero(int(id), type.lower(), key)
     try:
         collections = zot.collections_top()
@@ -47,7 +95,7 @@ def fetch_collections(id, type, key, messages):
 
 def select_collection(collection, messages):
     if collection is None:
-        return None, messages, gr.HTML.update()
+        return None, messages, gr.HTML.update(), None
     collection_name = re.sub('\s\(\d+\)$', '', collection)
     messages.set([Message(
         Icons.OK, f"Selected collection: <span style='font-weight: bold'>{collection_name}</span>. Please type your question and hit \"Enter\".")])