Update app.py

app.py

@@ -13,9 +13,15 @@ if not os.path.exists(app.config["UPLOAD_FOLDER"]):
 with open("demo_answers.json") as f:
     answers = json.load(f)
 
+analysis_requested = False
+current_analysis_key = None
+
 
 @app.route("/")
 def index():
+    global analysis_requested, current_analysis_key
+    analysis_requested = False
+    current_analysis_key = None
     return render_template("index.html")
 
 
@@ -44,7 +50,22 @@ def upload_file():
 
 @app.route("/chat", methods=["POST"])
 def chat():
+    global analysis_requested, current_analysis_key
     user_message = request.json.get("message").lower()
+
+    if analysis_requested and user_message == "yes":
+        return add_insight()
+
+    if user_message == "proceed with the analysis of the data":
+        analysis_requested = True
+        current_analysis_key = "01"
+        response_text = (
+            "Okay, let's start the analysis. "
+            + answers[current_analysis_key]["answer"]["answer"]
+            + " Would you like to add this insight to the dashboard?"
+        )
+        return jsonify({"response": response_text})
+
     # Find the answer by matching the user message with questions in the JSON file
     matched_entry = next(
         (
@@ -107,5 +128,60 @@ def explore():
     )
 
 
+@app.route("/add_insight", methods=["POST"])
+def add_insight():
+    global analysis_requested, current_analysis_key
+    if analysis_requested and current_analysis_key:
+        key = current_analysis_key
+        matched_entry = answers.get(key)
+
+        if matched_entry:
+            response = matched_entry["answer"]["answer"]
+            justification = matched_entry["answer"]["justification"]
+            image_filename = f"{key}.jpg"
+            image_path = os.path.join(app.config["IMAGES_FOLDER"], image_filename)
+
+            if os.path.exists(image_path):
+                image_url = url_for("static", filename=f"images/{image_filename}")
+            else:
+                image_url = None
+
+            next_key = str(int(key) + 1).zfill(2)
+            if next_key in answers:
+                current_analysis_key = next_key
+                next_response = (
+                    answers[next_key]["answer"]["answer"]
+                    + " Would you like to add this insight to the dashboard?"
+                )
+                return jsonify(
+                    {
+                        "response": response,
+                        "justification": justification,
+                        "image_url": image_url,
+                        "next_response": next_response,
+                        "key": key,
+                    }
+                )
+            else:
+                current_analysis_key = None
+                return jsonify(
+                    {
+                        "response": response,
+                        "justification": justification,
+                        "image_url": image_url,
+                        "key": key,
+                        "next_response": None,
+                    }
+                )
+        else:
+            return jsonify({"response": "No more data to explore."})
+    else:
+        return jsonify(
+            {
+                "response": "Please request to proceed with the analysis of the data first."
+            }
+        )
+
+
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", debug=True, port=7860)
+    app.run(debug=True, port=7860, host="0.0.0.0")