modified main.py

main.py

@@ -61,6 +61,14 @@ def get_model_response(model , input_text):
     chain = final_prompt | chat | output_parser
     return chain.invoke({"input": input_text})
 
+def format_response(response: str) -> str:
+    # Replace '\n\n' with two new lines to keep paragraph breaks
+    formatted_response = response.replace('\\n\\n', '\n\n')
+    
+    # Replace '\\n' with a single new line to maintain line breaks
+    formatted_response = formatted_response.replace('\\n', '\n')
+    
+    return formatted_response
 
 @app.post("/generate-response/")
 async def generate_response(request: PromptRequest):
@@ -68,7 +76,7 @@ async def generate_response(request: PromptRequest):
     model_choice = request.model.lower()
 
     response = get_model_response(model_choice , input_text)
-    return {"response": response}
+    return {"response": format_response(response)}
 
 # To run the FastAPI app, use:
 # uvicorn main:app --reload