testing csv export on railway

ai_ta_backend/export_data.py

@@ -2,6 +2,7 @@
 import datetime
 import pandas as pd
 import supabase
+from flask import send_file
 
 def export_convo_history_csv(course_name: str, from_date= '', to_date= ''):
     """
@@ -35,6 +36,7 @@ def export_convo_history_csv(course_name: str, from_date= '', to_date= ''):
         print("id count greater than zero")
         first_id = response.data[0]['id']
         last_id = response.data[-1]['id']
+        filename = course_name + '_convo_history.csv'
 
         # Fetch data in batches of 25 from first_id to last_id
         while first_id <= last_id:
@@ -43,17 +45,23 @@ def export_convo_history_csv(course_name: str, from_date= '', to_date= ''):
             # Convert to pandas dataframe
             df = pd.DataFrame(response.data)
             # Append to csv file
-            filename = course_name + '_convo_history.csv'
-            if not os.path.isfile(filename):
-                df.to_csv(filename, mode='a', header=True, index=False)
+            if not os.path.isfile('docs/' + filename):
+                df.to_csv('docs/' + filename, mode='a', header=True, index=False)
             else:
-                df.to_csv(filename, mode='a', header=False, index=False)
+                df.to_csv('docs/' + filename, mode='a', header=False, index=False)
 
             # Update first_id
             first_id = response.data[-1]['id'] + 1
             print("updated first_id: ", first_id)
-
+
+        # Download file
+        file_path = os.path.join(os.getcwd(), 'docs/' + filename)
+        try:
+            return (file_path, 'docs/' + filename, os.getcwd())
+        except Exception as e:
+            print(e)
+            return "Error downloading file"
     else:
         print("No data found between the dates")
         return "No data found between the dates"
-    return "success"
+

ai_ta_backend/main.py

@@ -5,7 +5,7 @@
 from typing import List
 
 from dotenv import load_dotenv
-from flask import Flask, Response, abort, jsonify, request
+from flask import Flask, Response, abort, jsonify, request, send_file, make_response, send_from_directory
 from flask_cors import CORS
 from flask_executor import Executor
 from sqlalchemy import JSON
@@ -486,9 +486,15 @@ def export_convo_history():
     )
 
   export_status = export_convo_history_csv(course_name, from_date, to_date)
-
-  response = jsonify(export_status)
+  print(export_status)
+  print("path: ", os.getcwd() + '/docs/cropwizard_convo_history.csv')
+  print("path exists: ", os.path.exists(os.getcwd() + '/docs/cropwizard_convo_history.csv'))
+
+  response = make_response(send_file(os.getcwd() + '/docs/cropwizard_convo_history.csv', as_attachment=True))
+  #response = make_response(send_from_directory('docs', 'cropwizard_convo_history.csv', as_attachment=True))
   response.headers.add('Access-Control-Allow-Origin', '*')
+  response.headers["Content-Disposition"] = f"attachment; filename={export_status[1]}"
+
   return response