init

app.py

@@ -3,9 +3,9 @@ import openai
 import numpy as np
 
 
-def calc_distance(text1, text2) -> str:
-    text1_emb = openai.Embedding.create(input=text1, model="text-embedding-ada-002")
-    text2_emb = openai.Embedding.create(input=text2, model="text-embedding-ada-002")
+def calc_distance(text1, text2, api) -> str:
+    text1_emb = openai.Embedding.create(input=text1, model="text-embedding-ada-002", api_key=api)
+    text2_emb = openai.Embedding.create(input=text2, model="text-embedding-ada-002", api_key=api)
 
     text1_emb = np.array(text1_emb)
     text2_emb = np.array(text2_emb)
@@ -15,6 +15,8 @@ def calc_distance(text1, text2) -> str:
 
 
 with gr.Blocks() as b:
+    openai_apikey = gr.TextArea(label="OpenAI API key", lines=1)
+
     with gr.Row():
         with gr.Column():
             text1 = gr.TextArea(label="Text 1")
@@ -23,7 +25,7 @@ with gr.Blocks() as b:
 
         output = gr.outputs.Label(label="Distance")
 
-    btn.click(inputs=(text1, text2), outputs=output)
+    btn.click(fn=calc_distance ,inputs=(text1, text2, openai_apikey), outputs=output)
 
 
 b.launch()