Update main.py

main.py

@@ -24,6 +24,7 @@ from langchain_core.prompts import ChatPromptTemplate
 from datetime import datetime
 from presentation_api import router as presentation_router
 from yf_docs import yf_docs
+import pandas as pd
 
 app = FastAPI()
 app.include_router(presentation_router)
@@ -102,10 +103,12 @@ save_yf_prompt = """
 # Save downloaded data
 data = yf.download(['MSFT', 'AAPL', 'GOOG'], period='6mo')
 data.to_pickle('stock_data.pkl')
-"""
+
 # Load it back later
 data = pd.read_pickle('stock_data.pkl')
 
+"""
+
 memory = MemorySaver()
 model = ChatOpenAI(model="gpt-4o", streaming=True)
 prompt = ChatPromptTemplate.from_messages([