Update app.py

app.py

@@ -1,10 +1,7 @@
 # app.py
 import gradio as gr
-from questions import questions
-from chain_reports import generate_short_report_for_session
-from chain_problems import analyze_problems_with_chain
-from chain_recommendations import generate_recommendations
-from chain_summary import generate_final_summary, shorten_summary
+from pipeline import process_answers_pipeline  # Import the centralized pipeline function
+from questions import questions  # Import questions list
 
 def process_answers(
     sleep: str,
@@ -18,7 +15,7 @@ def process_answers(
     manage_stress: str,
     routine: str
 ):
-    # Map user inputs to questions
+    # Map user inputs to corresponding questions
     responses = {
         questions[0]: sleep,
         questions[1]: exercise,
@@ -32,35 +29,37 @@ def process_answers(
         questions[9]: routine
     }
 
-    report = generate_short_report_for_session(responses)
-    problems = analyze_problems_with_chain(responses, report)
-    recommendation = generate_recommendations(problems)
-    final_summary = generate_final_summary(report, problems, recommendation)
-    shortened_summary = shorten_summary(final_summary)
+    # Use the centralized pipeline to process all responses
+    results = process_answers_pipeline(responses)
+
+    # Format outputs using results from the pipeline
+    wellness_report = f"**Wellness Report**\n------------------\n{results['report'].strip()}"
 
-    wellness_report = f"**Wellness Report**\n------------------\n{report.strip()}"
     identified_problems = (
         "**Identified Problems**\n"
         "-----------------------\n"
-        f"Sleep Problem: {problems.get('sleep_problem', 'N/A')}%\n"
-        f"Exercise Problem: {problems.get('exercise_problem', 'N/A')}%\n"
-        f"Stress Problem: {problems.get('stress_problem', 'N/A')}%\n"
-        f"Diet Problem: {problems.get('diet_problem', 'N/A')}%"
+        f"Sleep Problem: {results['problems'].get('sleep_problem', 'N/A')}%\n"
+        f"Exercise Problem: {results['problems'].get('exercise_problem', 'N/A')}%\n"
+        f"Stress Problem: {results['problems'].get('stress_problem', 'N/A')}%\n"
+        f"Diet Problem: {results['problems'].get('diet_problem', 'N/A')}%"
     )
+
     recommendations = (
         "**Recommendations**\n"
         "--------------------\n"
-        f"{recommendation.strip()}"
+        f"{results['recommendation'].strip()}"
     )
+
     summary_shown = (
         "**Summary (SHOWN TO USER)**\n"
         "-----------------\n"
-        f"{final_summary.strip()}"
+        f"{results['final_summary'].strip()}"
     )
+
     final_summary_video = (
         "**Final Summary (FOR VIDEO CREATION)**\n"
         "-----------------\n"
-        f"{shortened_summary.strip()}"
+        f"{results['shortened_summary'].strip()}"
     )
 
     return wellness_report, identified_problems, recommendations, summary_shown, final_summary_video