using env file

_.env

@@ -0,0 +1 @@
+OPENAI_API_KEY=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

app.py

@@ -187,27 +187,17 @@ def get_completion(conversation, model="gpt-4-1106-preview"):
 
     [{sample_output}]
     """
-
-    # load_dotenv(find_dotenv("_.env"), override=True)
-
-    openai_api_key = os.environ['OPENAI_API_KEY']
-
-    if openai_api_key is None:
-        raise ValueError("No OpenAI API key found. Please set the OPENAI_API_KEY environment variable.")
     
+    client = OpenAI()
+    
+    messages = [{"role": "user", "content": prompt}]
+    response = client.chat.completions.create(
+        model=model,
+        messages=messages,
+        temperature=0, # this is the degree of randomness of the model's output
+    )
 
-    openai.api_key = openai_api_key
-
-    # client = OpenAI()
-
-    # messages = [{"role": "user", "content": prompt}]
-    # response = client.chat.completions.create(
-    #     model=model,
-    #     messages=messages,
-    #     temperature=0, # this is the degree of randomness of the model's output
-    # )
-
-    # analysis = response.choices[0].message.content
+    analysis = response.choices[0].message.content
 
     def extract_conv_with_labels(analysis):
         analysis = analysis.replace("\n", " ")
@@ -300,10 +290,16 @@ def get_completion(conversation, model="gpt-4-1106-preview"):
 
         return fig
     
-    # fig = sentiment_flow_plot(analysis)
+    fig = sentiment_flow_plot(analysis)
 
-    # return response.choices[0].message.content, fig
-    return openai_api_key
+    return response.choices[0].message.content, fig
+
+def set_key(key):
+    with open("_.env", "w") as file:
+        file.write(f"OPENAI_API_KEY={key}")
+        
+    load_dotenv(find_dotenv("_.env"), override=True)
+    return
 
 import gradio as gr
 
@@ -315,7 +311,9 @@ with gr.Blocks() as gpt_analysis:
         volunteer's perspective.<br /> Click on them and submit them to the model to see how it works.")
     conversation = gr.Textbox(label="Input", lines=2)
     btn = gr.Button(value="Submit")
-
+    api_key = gr.Textbox(label="Key", lines=1)
+    btn_key = gr.Button(value="Submit Key")
+    btn_key.click(set_key, inputs=api_key)
     with gr.Row():
         output_box = gr.Textbox(value="", label="Output",lines=4)
         plot_box = gr.Plot(label="Analysis Plot")