server/app.py

@@ -1,4 +1,5 @@
 import os
+import datetime
 
 from dotenv import load_dotenv
 from flask import Flask, jsonify, request
@@ -42,7 +43,6 @@ auth_manager = AuthManager(db, bcrypt)
 
 stat_gen = StatGen()
 
-
 @app.route("/register", methods=["POST"])
 def register_user():
     data = request.get_json()
@@ -112,7 +112,7 @@ def upload_data():
     post_file = request.files["posts"]
     topic_file = request.files["topics"]
 
-    if post_file.filename == "" or topic_file == "":
+    if post_file.filename == "" or topic_file.filename == "":
         return jsonify({"error": "Empty filename"}), 400
 
     if not post_file.filename.endswith(".jsonl") or not topic_file.filename.endswith(
@@ -280,75 +280,34 @@ def get_interaction_analysis(dataset_id):
         return jsonify({"error": f"An unexpected error occurred: {str(e)}"}), 500
 
 
-# @app.route("/filter/query", methods=["POST"])
+@app.route("/dataset/query", methods=["POST"])
-# def filter_query():
+@jwt_required()
-#     if stat_obj is None:
+def filter_query():
-#         return jsonify({"error": "No data uploaded"}), 400
+    data = request.get_json()
 
-#     data = request.get_json(silent=True) or {}
+    if "query" in data:
+        stat_gen.set_search_query(data["query"])
 
-#     if "query" not in data:
+    if "start" in data:
-#         return jsonify(stat_obj.df.to_dict(orient="records")), 200
+        start_timestamp = datetime.datetime.fromisoformat(data["start"])
+        stat_gen.set_start_date(start_timestamp)
 
-#     query = data["query"]
+    if "end" in data:
-#     filtered_df = stat_obj.filter_by_query(query)
+        end_timestamp = datetime.datetime.fromisoformat(data["end"])
+        stat_gen.set_end_date(end_timestamp)
 
-#     return jsonify(filtered_df), 200
+    if "sources" in data:
+        data_sources = set(data["sources"])
+        stat_gen.set_data_sources(data_sources)
+
+    return jsonify({"message": "Filters set successfully"}), 200
 
 
-# @app.route("/filter/time", methods=["POST"])
+@app.route("/database/query/reset", methods=["GET"])
-# def filter_time():
+@jwt_required()
-#     if stat_obj is None:
+def reset_dataset():
-#         return jsonify({"error": "No data uploaded"}), 400
+    stat_gen.reset_filters()
-
+    return jsonify({"message": "Filters reset successfully"}), 200
-#     data = request.get_json(silent=True)
-#     if not data:
-#         return jsonify({"error": "Invalid or missing JSON body"}), 400
-
-#     if "start" not in data or "end" not in data:
-#         return jsonify({"error": "Please include both start and end dates"}), 400
-
-#     try:
-#         start = pd.to_datetime(data["start"], utc=True)
-#         end = pd.to_datetime(data["end"], utc=True)
-#         filtered_df = stat_obj.set_time_range(start, end)
-#         return jsonify(filtered_df), 200
-#     except Exception:
-#         return jsonify({"error": "Invalid datetime format"}), 400
-
-
-# @app.route("/filter/sources", methods=["POST"])
-# def filter_sources():
-#     if stat_obj is None:
-#         return jsonify({"error": "No data uploaded"}), 400
-
-#     data = request.get_json(silent=True)
-#     if not data:
-#         return jsonify({"error": "Invalid or missing JSON body"}), 400
-
-#     if "sources" not in data:
-#         return jsonify({"error": "Ensure sources hash map is in 'sources' key"}), 400
-
-#     try:
-#         filtered_df = stat_obj.filter_data_sources(data["sources"])
-#         return jsonify(filtered_df), 200
-#     except ValueError:
-#         return jsonify({"error": "Please enable at least one data source"}), 400
-#     except Exception as e:
-#         return jsonify({"error": "An unexpected server error occured: " + str(e)}), 500
-
-
-# @app.route("/filter/reset", methods=["GET"])
-# def reset_dataset():
-#     if stat_obj is None:
-#         return jsonify({"error": "No data uploaded"}), 400
-
-#     try:
-#         stat_obj.reset_dataset()
-#         return jsonify({"success": "Dataset successfully reset"})
-#     except Exception as e:
-#         print(traceback.format_exc())
-#         return jsonify({"error": f"An unexpected error occurred: {str(e)}"}), 500
 
 
 if __name__ == "__main__":

server/stat_gen.py

@@ -127,14 +127,16 @@ class StatGen:
     def set_search_query(self, search_query: str) -> None:
         self.search_query = search_query
 
-    def set_time_range(self, start: datetime.datetime, end: datetime.datetime) -> None:
+    def set_start_date(self, start: datetime.datetime) -> None:
         self.start_date_filter = start
+
+    def set_end_date(self, end: datetime.datetime) -> None:
         self.end_date_filter = end
 
-    def search_data_sources(self, data_sources: set) -> None:
+    def set_data_sources(self, data_sources: set) -> None:
         self.data_source_filters = data_sources
         
-    def reset_dataset(self) -> None:
+    def reset_filters(self) -> None:
         self.search_query = ""
         self.start_date_filter = None
         self.end_date_filter = None