clean up main for only simple stream app

app_simple_stream/app/main.py

@@ -1,8 +1,8 @@
 from fastapi import FastAPI, Request
 from langchain_core.runnables import Runnable
-from langchain_core.callbacks import BaseCallbackHandler # Implement Server-Sent Event protocol
+from langchain_core.callbacks import BaseCallbackHandler # Print what is sent to llm and what was recieved as answer
 
-from sse_starlette.sse import EventSourceResponse
+from sse_starlette.sse import EventSourceResponse # Implement Server-Sent Event protocol
 from langserve.serialization import WellKnownLCSerializer # Encode data because we are using Langserve on the frontend to recieve events, and Langserve is using this class to decode the data
 from typing import List
 
@@ -17,7 +17,8 @@ async def generate_stream(input_data: schemas.BaseModel, runnable: Runnable, cal
     for output in runnable.stream(input_data.dict(), config={"callbacks": callbacks}):
         data = WellKnownLCSerializer().dumps(output).decode("utf-8") # Decode data
         yield {'data': data, "event": "data"} 
-        yield {"event": "end"}
+    
+    yield {"event": "end"}
 
 
 @app.post("/simple/stream")