Update app.py

app.py

@@ -17,20 +17,16 @@ os.environ["OPENAI_API_KEY"] = "6c6a1a46606c45f2b7b67d8796cf068a"
 # Import Azure OpenAI
 from langchain.llms import AzureOpenAI
 
-# Create an instance of Azure OpenAI
-# Replace the deployment name with your own
-llm = AzureOpenAI(
-    deployment_name="gpt-35-turbo",
-    model_name="chatgpt",
-)
-
+form = st.sidebar.form(key='my_form')
+user_api_key = form.text_input(
+    label="#### Your OpenAI API key 👇",
+    placeholder="Paste your openAI API key, sk-",
+    type="password")
 uploaded_file = form.file_uploader("Choose a file")
 text_prompt = form.text_area('Text Prompt', value='Hello, how are you?')
 submit_button = form.form_submit_button(label='Submit')
 
 
-
-
 if submit_button :
     df = pd.read_csv(uploaded_file)
 
@@ -46,15 +42,16 @@ if submit_button :
 
     st.subheader("AI generated response:")
 
-    
-    # Send a message to the model
-    response = llm("Tell me a joke")
-    
-    # Print the response
-    print(response)
+
+    llm = AzureOpenAI(
+        deployment_name="chatgpt",
+        model_name="chatgpt",
+        openai_api_key=user_api_key,
+    )
+
     
     agent = create_pandas_dataframe_agent(
-        ChatOpenAI(temperature=0, model="gpt-3.5-turbo-16k", openai_api_key=user_api_key),
+        llm,
         df,
         verbose=True,
         agent_type=AgentType.OPENAI_FUNCTIONS,