Update app.py

app.py

@@ -31,7 +31,12 @@ def upload_file():
         file.save(file_path)
         df = pd.read_csv(file_path)
         summary = df.describe(include="all").to_html(classes="table table-striped")
-        return jsonify({"summary": summary})
+        return jsonify(
+            {
+                "summary": summary,
+                "message": "File uploaded successfully. Let's see what we have here...",
+            }
+        )
     else:
         return jsonify({"error": "Unsupported file type"}), 400
 
@@ -71,5 +76,30 @@ def chat():
     )
 
 
+@app.route("/explore", methods=["POST"])
+def explore():
+    key = request.json.get("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
+    else:
+        response = "No more data to explore."
+        justification = None
+        image_url = None
+
+    return jsonify(
+        {"response": response, "justification": justification, "image_url": image_url}
+    )
+
+
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=True)
+    app.run(host="0.0.0.0", debug=True, port=7860)