Update app.py

app.py

@@ -23,7 +23,6 @@ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
 """
-
 from typing import Any
 import gradio as gr
 from langchain_openai import OpenAIEmbeddings
@@ -38,36 +37,21 @@ import fitz
 from PIL import Image
 import os
 import re
-import uuid
+import openai
 
 openai.api_key = "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"
 
-enable_box = gr.Textbox(
-    value=None, placeholder="Upload your OpenAI API key", interactive=True
-)
-disable_box = gr.Textbox(value="OpenAI API key is set", interactive=False)
-
-
-def set_apikey(api_key: str):
-    print("API Key set")
-    app.OPENAI_API_KEY = api_key
-    return disable_box
-
-
-def enable_api_box():
-    return enable_box
-
 
 def add_text(history, text: str):
     if not text:
-        raise gr.Error("enter text")
+        raise gr.Error("Enter text")
     history = history + [(text, "")]
     return history
 
 
-class my_app:
-    def __init__(self, OPENAI_API_KEY: str = None) -> None:
-        self.OPENAI_API_KEY: str = OPENAI_API_KEY
+class MyApp:
+    def __init__(self) -> None:
+        self.OPENAI_API_KEY: str = openai.api_key
         self.chain = None
         self.chat_history: list = []
         self.N: int = 0
@@ -145,19 +129,11 @@ def purge_chat_and_render_first(file):
     image = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
     return image, []
 
-app = my_app()
+
+app = MyApp()
 
 with gr.Blocks() as demo:
     with gr.Column():
-        with gr.Row():
-
-            with gr.Column(scale=1):
-                api_key = gr.Textbox(
-                    placeholder="Enter OpenAI API key and hit <RETURN>",
-                    show_label=False,
-                    interactive=True
-                )
-
         with gr.Row():
             with gr.Column(scale=2):
                 with gr.Row():
@@ -168,21 +144,13 @@ with gr.Blocks() as demo:
                         placeholder="Enter text and press submit",
                         scale=2
                     )
-                    submit_btn = gr.Button("submit", scale=1)
+                    submit_btn = gr.Button("Submit", scale=1)
 
             with gr.Column(scale=1):
                 with gr.Row():
                     show_img = gr.Image(label="Upload PDF")
                 with gr.Row():
-                    btn = gr.UploadButton("📁 upload a PDF", file_types=[".pdf"])
-
-    api_key.submit(
-        fn=set_apikey,
-        inputs=[api_key],
-        outputs=[
-            api_key,
-        ],
-    )
+                    btn = gr.UploadButton("📁 Upload a PDF", file_types=[".pdf"])
 
     btn.upload(
         fn=purge_chat_and_render_first,
@@ -204,4 +172,6 @@ with gr.Blocks() as demo:
     )
 
 demo.queue()
-demo.launch()
+demo.launch()
+
+