Update main.py

main.py

@@ -1,6 +1,7 @@
 import asyncio
 import aiohttp
 from fastapi import FastAPI
+from sse_starlette.sse import EventSourceResponse
 import logging
 
 logging.basicConfig(level=logging.INFO)
@@ -12,7 +13,7 @@ API_URL = "https://b917-160-179-178-105.ngrok-free.app/test-ai-call"
 
 call_count = 0
 
-async def call_api():
+async def event_generator():
     global call_count
     async with aiohttp.ClientSession() as session:
         while True:
@@ -24,18 +25,19 @@ async def call_api():
                     if response.status == 200:
                         data = await response.json()
                         logger.info(f"Call {call_count}: API Response: {data}")
+                        yield data
                     else:
                         logger.error(f"Call {call_count}: API call failed with status code: {response.status}")
 
             except Exception as e:
                 logger.error(f"Call {call_count}: Error: {e}")
 
-            await asyncio.sleep(1)
+            await asyncio.sleep(10)
 
-@app.on_event("startup")
-async def startup_event():
-    asyncio.create_task(call_api())
+@app.get("/stream")
+async def stream(request):
+    return EventSourceResponse(event_generator())
 
 @app.get("/")
 async def root():
-    return {"message": "API caller is running", "calls_made": call_count}
+    return {"message": "API caller is running", "calls_made": call_count}