Create chain_reports.py

chain_reports.py

@@ -0,0 +1,22 @@
+# chain_reports.py
+from typing import Dict
+from langchain import PromptTemplate, LLMChain
+from models import chat_model
+
+report_prompt_template = PromptTemplate(
+    input_variables=["qa_summary"],
+    template=(
+        "You are a wellness assistant. The user provided the following answers:\n\n"
+        "{qa_summary}\n\n"
+        "Based on these answers, provide a brief, actionable wellness report. "
+        "Include simple suggestions to improve their sleep, exercise, stress management, and diet. "
+        "Be concise and helpful.\n\n"
+        "Report:"
+    )
+)
+report_chain = LLMChain(llm=chat_model, prompt=report_prompt_template)
+
+def generate_short_report_for_session(responses: Dict[str, str]) -> str:
+    qa_summary = "\n".join(f"{q}: {a}" for q, a in responses.items())
+    raw_report = report_chain.run(qa_summary=qa_summary)
+    return raw_report.replace("*", "").replace("**", "")