force manual key entry if no env-var provided

app.py

@@ -4,19 +4,24 @@ from pandas import DataFrame
 
 from pedalo.main import run
 import pandas as pd
+import os
+import openai
 
 # st.set_page_config(layout="wide")
 st.title("PEDALO - Productive Exploratory Data Analysis using Langchain interrOgation")
 st.write("Ask your data what you wanna know!")
+if "OPENAI_API_KEY" in os.environ:
+    openai_api_key = os.environ.get("OPENAI_API_KEY")
+else:
+    openai_api_key = st.sidebar.text_input("OPENAI_API_KEY")
 model = st.sidebar.radio("Which model do you wanna use?", ("gpt-4", "gpt-3.5-turbo"), index=1)
-
 uploaded_file = st.sidebar.file_uploader("Choose a file", type=["csv"])
 
 
 
 def run_df_analysis(prompt:str, df: DataFrame):
     st_callback = StreamlitCallbackHandler(st.container())
-    response = run(prompt, df, st_callback, model)
+    response = run(prompt, df, st_callback, openai_api_key, model)
     st.write(response)
 
 

pedalo/agents/code_interpreter.py

@@ -12,7 +12,7 @@ model_grand_agent = "gpt-3.5-turbo"
 
 
 def run(
-    prompt: str, df: DataFrame, st_callback: StreamlitCallbackHandler, model="gpt-4"
+    prompt: str, df: DataFrame, st_callback: StreamlitCallbackHandler, openai_api_key:str, model="gpt-4"
 ) -> str:
     agent_executor_kwargs = {
         "handle_parsing_errors": True,
@@ -22,6 +22,7 @@ def run(
             temperature=0,
             model=model_pandas_agent,
             streaming=True,
+            openai_api_key=openai_api_key
         ),
         df=df,
         verbose=True,
@@ -47,7 +48,7 @@ def run(
                                                takes as an input the entire question and returns the answer after running calculations""",
             ),
         ],
-        llm=ChatOpenAI(temperature=0, model=model_grand_agent, streaming=True),
+        llm=ChatOpenAI(temperature=0, model=model_grand_agent, streaming=True, openai_api_key=openai_api_key),
         agent_type=AgentType.OPENAI_FUNCTIONS,
         # agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
         verbose=True,

pedalo/main.py

@@ -13,9 +13,9 @@ openai.api_key = api_key
 
 
 def run(
-    prompt: str, df: DataFrame, st_callback: StreamlitCallbackHandler, model="gpt-4"
+    prompt: str, df: DataFrame, st_callback: StreamlitCallbackHandler, openai_api_key: str, model="gpt-4"
 ):
-    result = code_interpreter.run(prompt, df, st_callback, model)
+    result = code_interpreter.run(prompt, df, st_callback, openai_api_key, model)
     return result