Update app.py

app.py

@@ -17,7 +17,7 @@ from deepdoctection.utils.settings import get_type
 from dd_addons.analyzer.loader import get_loader
 from dd_addons.extern.guidance import TOKEN_DEFAULT_INSTRUCTION
 from dd_addons.utils.settings import register_llm_token_tag, register_string_categories_from_list
-from dd_addons.extern.openai import OpenAiLmmTokenClassifier
+from dd_addons.extern.openai import OpenAiLmmTokenClassifier, is_api_key_valid
 
 import gradio as gr
 
@@ -27,6 +27,8 @@ demo = gr.Blocks(css="scrollbar.css")
 
 
 def process_analyzer(openai_api_key, categories_str, instruction_str, img, pdf, max_datapoints):
+    if not is_api_key_valid(openai_api_key):
+        return [], {}, "You have entered no or an invalid api key. Please enter a valid api key"
     categories_list = categories_str.split(",")
     register_string_categories_from_list(categories_list, "custom_token_classes")
     custom_token_class = dd.object_types_registry.get("custom_token_classes")
@@ -65,7 +67,7 @@ def process_analyzer(openai_api_key, categories_str, instruction_str, img, pdf,
         json_out[f"page_{idx}"] = dp.get_token()
 
     return [dp.viz(show_cells=False, show_layouts=False, show_tables=False, show_words=True, show_token_class=True, ignore_default_token_class=True)
-            for dp in dpts], json_out
+            for dp in dpts], json_out, "No error"
 
 
 with demo:
@@ -93,6 +95,7 @@ with demo:
                 with gr.Box():
                     gr.Markdown("Enter your OpenAI API Key* ")
                     user_token = gr.Textbox(value='', placeholder="OpenAI API Key", type="password", show_label=False)
+
                     gr.Markdown("<sup>* Your API key will not be saved. However, it is always recommended to deactivate the"
                                 "API key once it is entered into an unknown source</sup>")
             with gr.Column():
@@ -100,7 +103,7 @@ with demo:
                     gr.Markdown(
                         "Enter a list of comma seperated entities. Use a snake case style. Avoid special characters. "
                         "Best way is to only use `a-z` and `_`")
-                    categories = gr.Textbox(value='mitarbeiter_anzahl', placeholder="mitarbeiter_anzahl", show_label=False)
+                    categories = gr.Textbox(value='', placeholder="mitarbeiter_anzahl", show_label=False)
                 with gr.Box():
                     gr.Markdown("Optional: Enter a prompt for additional guidance. Will use the placeholder as fallback")
                     instruction = gr.Textbox(value='', placeholder=TOKEN_DEFAULT_INSTRUCTION, show_label=False)
@@ -114,6 +117,10 @@ with demo:
     with gr.Box():
         gr.Markdown("<h2><center>Outputs</center></h2>")
         with gr.Row():
+            with gr.Column():
+                with gr.Box():
+                    gr.Markdown("<center><strong>Message</strong></center>")
+                    msg = gr.Textbox(value='', placeholder="message", show_label=False)
             with gr.Column():
                 with gr.Box():
                     gr.Markdown("<center><strong>JSON</strong></center>")
@@ -130,6 +137,6 @@ with demo:
                 html = gr.HTML()
 
     btn.click(fn=process_analyzer, inputs=[user_token, categories,  instruction, inputs, inputs_pdf, max_imgs],
-              outputs=[gallery, json])
+              outputs=[gallery, json, msg])
 
 demo.launch()