Update main.py

main.py

@@ -7,8 +7,8 @@ import time
 import os
 
 dotenv.load_dotenv()
-API_KEY = os.getenv('API_KEY')
-ACCESS_KEY = os.getenv('ACCESS_KEY')
+API_KEY = os.getenv("API_KEY")
+ACCESS_KEY = os.getenv("ACCESS_KEY")
 
 def create_client(api_key):
     client = openai.OpenAI(
@@ -22,10 +22,10 @@ def generate_summary(client, corpus):
     math_summary = generate_math_summary(corpus)
     if not math_summary:
         print("Error generating Math Summary")
-        response['summary_status'] = "error"
-        response['summary'] = None
-        response['mindmap_status'] = "error"
-        response['mindmap'] = None
+        response["summary_status"] = "error"
+        response["summary"] = None
+        response["mindmap_status"] = "error"
+        response["mindmap"] = None
         return response
     else:
         response = generate_nlp_summary_and_mindmap(client, corpus)