Merge pull request #3 from yashksaini-coder/kush

Main python server file update

Author: yashksaini-coder

app.py

@@ -8,7 +8,8 @@
 import os
 from dotenv import load_dotenv
 # Load API key from .env file
-
+from fastapi.middleware.cors import CORSMiddleware
+from topStocks import get_top_stocks
 load_dotenv(dotenv_path=".env")
 GROQ_API_KEY = os.getenv("api_key")
 
@@ -19,6 +20,14 @@
 
 app = FastAPI()
 # Web searching agent
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"], 
+    allow_credentials=True,
+    allow_methods=["*"],  
+    allow_headers=["*"],  
+)
+
 web_search_agent = Agent(
     name="web_agent",
     role="search the web for information based on the user given input",
@@ -66,6 +75,12 @@
     markdown=True,
 )
 
+@app.get("/top-stocks")
+async def read_top_stocks():
+    top_stocks = ['AAPL', 'GOOGL', 'MSFT', 'AMZN', 'TSLA']
+    stock_info = get_top_stocks(top_stocks)
+    return stock_info
+
 @app.get("/ask")
 def ask(query: str):
     """

main.py

@@ -2,8 +2,7 @@
 import groq
 import os
 from dotenv import load_dotenv
-from fastapi.middleware.cors import CORSMiddleware
-from topStocks import get_top_stocks
+
 # Load API key from .env file
 
 load_dotenv(dotenv_path=".env")
@@ -17,13 +16,7 @@
 
 app = FastAPI()
 
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"], 
-    allow_credentials=True,
-    allow_methods=["*"],  
-    allow_headers=["*"],  
-)
+
 
 @app.get("/ask")
 def ask(query: str):
@@ -45,9 +38,3 @@ def ask(query: str):
 
     except Exception as e:
         return {"error": str(e)}
-
-@app.get("/top-stocks")
-async def read_top_stocks():
-    top_stocks = ['AAPL', 'GOOGL', 'MSFT', 'AMZN', 'TSLA']
-    stock_info = get_top_stocks(top_stocks)
-    return stock_info