Update app.py

app.py

@@ -2,15 +2,18 @@ import os
 import openai
 from io import StringIO 
 import sys
-import csv
+
 import gradio as gr
 from langchain.agents import create_csv_agent
 from langchain.chat_models import AzureChatOpenAI
-
+from langchain.llms import AzureOpenAI
 
 os.environ["OPENAI_API_TYPE"] = openai.api_type = "azure"
 os.environ["OPENAI_API_BASE"] = openai.api_base = "https://openai-endpoint.openai.azure.com/"
 os.environ["OPENAI_API_KEY"] = openai.api_key = os.environ["OPENAI_API_KEY"]
+openai.api_version = "2023-03-15-preview"
+
+llm = AzureOpenAI(deployment_name="davinci003", model_name="text-davinci-003")
 
 class Capturing(list):
     def __enter__(self):
@@ -24,8 +27,10 @@ class Capturing(list):
 
 def answer_question(input_file, question):
     agent = create_csv_agent(llm, input_file.name, verbose=True)
+    question = question
     with Capturing() as printed_text:
         answer = agent.run(question)
+       
     import re
     text = '\n'.join(printed_text) + '\n' + str(answer)
     # Remove all escape characters
@@ -38,9 +43,6 @@ def answer_question(input_file, question):
     text = text.strip()
     return text
 
-
-llm = AzureChatOpenAI(openai_api_base="https://openai-endpoint.openai.azure.com/", openai_api_key=openai.api_key, deployment_name="gpt4", model_name="gpt-4", openai_api_version="2023-03-15-preview")
-
 with gr.Blocks(css="footer {visibility: hidden}", title="CSV Copilot") as demo:
     csv_file = gr.State([])
     csv_file = gr.File(label="CSV File", accept=".csv")