Update app.py

app.py

@@ -11,7 +11,7 @@ from langchain.llms import AzureOpenAI
 
 # os.environ["OPENAI_API_TYPE"] = openai.api_type = "azure"
 # os.environ["OPENAI_API_BASE"] = openai.api_base = "https://aoai-southcentral.openai.azure.com/"
-os.environ["OPENAI_API_KEY"] = openai.api_key = "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"
+openai.api_key = "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"
 # openai.api_version = "2023-03-15-preview"
 
 
@@ -27,11 +27,14 @@ class Capturing(list):
         sys.stdout = self._stdout
 
 def answer_question(input_file, question):
-    llm = ChatOpenAI(model_name="gpt-4")
+    llm = ChatOpenAI(temperature=0)
     agent = create_csv_agent(llm, input_file.name, verbose=True)
     question = question
     with Capturing() as printed_text:
-        answer = agent.run(question)
+        try:
+            answer = agent.run(question)
+        except Exception as e:
+            answer = "LLM: " + str(e)
        
     import re
     text = '\n'.join(printed_text) + '\n' + str(answer)