app create

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+from openai import OpenAIError
+
 from selfinstruct.instance import instance_main
 import openai
 from langchain.llms.openai import OpenAI
@@ -9,15 +11,17 @@ def OpenaiChat(key, input, temperature):
     print(input)
     openai.api_key = key
     MODEL = "gpt-3.5-turbo"
-    response = openai.ChatCompletion.create(
-        model=MODEL,
-        messages=[
-            {"role": "user", "content": input}
-        ],
-        temperature=temperature
-    )
-    return json.dumps(response["choices"], ensure_ascii=False)
-
+    try:
+        response = openai.ChatCompletion.create(
+            model=MODEL,
+            messages=[
+                {"role": "user", "content": input}
+            ],
+            temperature=temperature
+        )
+        return json.dumps(response["choices"], ensure_ascii=False)
+    except OpenAIError:
+        return "Wrong Key!"
 
 with gr.Blocks() as demo:
     gr.Markdown("# Self-instruct Prompt")

selfinstruct/instance.py

@@ -6,6 +6,9 @@ import re
 import argparse
 import pandas as pd
 from collections import OrderedDict
+
+from openai import OpenAIError
+
 from .gpt3_api import make_requests as make_gpt3_requests
 from .templates.instance_gen_template import output_first_template_for_clf, input_first_template_for_gen
 from .templates.clf_task_template import template_1
@@ -302,9 +305,23 @@ def prepare_finetune(inputs, api_key):
     return results1, classify_res, instance_outputs, results2
 
 def instance_main(inputs, key):
+    try:
+        import openai
+        openai.api_key = key
+        MODEL = "gpt-3.5-turbo"
+        openai.ChatCompletion.create(
+            model=MODEL,
+            messages=[
+                {"role": "user", "content": "Hi"}
+            ],
+            temperature=1
+        )
+    except OpenAIError:
+        return "Wrong Key!", " ", " ", " "
     api_key = key
     inputs = inputs.split('\n')
     print("***", inputs)
     return prepare_finetune(inputs, api_key)
 
+
 # instance_main()