Update main.py

main.py

@@ -3,6 +3,7 @@ from pydantic import BaseModel
 import json
 import openai
 from typing import List, Dict, Any, Optional
+import os
 
 app = FastAPI()
 
@@ -162,7 +163,7 @@ async def generate_roadmap(input_data: RoadmapInput):
         MAKE SURE THAT WE INCLUDE EACH SUBTOPIC OF EACH CHAPTER FROM {physics_topics},{chemistry_topics} and {mathematics_topics} IS FINISHED
         YOU ARE NOT CONSTRAINED TO CREATE A ROADMAP FOR ONLY 'X' NUMBER OF DAYS , YOU CAN EXTEND TILL THE TOPICS ARE FINISHED BUT ONLY STICK TO THE TIMEFRAME ALLOTED FOR EACH SUBJECT AND DO NOT GO ABOVE OR BELOW THAT TIME FRAME.
         """
-
+        openai.api_key=os.getenv("KEY")
         # First LLM call
         response = openai.ChatCompletion.create(
             model="gpt-4o-mini",