Full Download

app.py

@@ -4,7 +4,9 @@ Provides endpoints for database operations on prompts and responses.
 """
 
 import os
-from flask import Flask, request, jsonify
+import csv
+import io
+from flask import Flask, request, jsonify, Response
 from flask_cors import CORS
 from dotenv import load_dotenv
 from database import Database
@@ -545,6 +547,65 @@ def health_check():
         return jsonify({'success': False, 'error': str(e)}), 500
 
 
+# ========== CSV DOWNLOAD ENDPOINTS ==========
+
+@app.route('/api/download/csv', methods=['GET'])
+def download_csv():
+    """
+    Download CSV files for prompts and responses tables.
+    Returns a zip file or individual CSV files.
+    """
+    try:
+        # Get all prompts
+        prompts = db.get_all_prompts(limit=None, offset=0)
+        
+        # Get all responses
+        responses = db.get_all_responses(limit=None, offset=0)
+        
+        # Create CSV for prompts
+        prompts_output = io.StringIO()
+        prompts_writer = csv.writer(prompts_output)
+        prompts_writer.writerow(['ID', 'Text', 'Note'])
+        for prompt in prompts:
+            prompts_writer.writerow([
+                prompt.get('id', ''),
+                prompt.get('text', ''),
+                prompt.get('note', '') or ''
+            ])
+        prompts_csv = prompts_output.getvalue()
+        
+        # Create CSV for responses
+        responses_output = io.StringIO()
+        responses_writer = csv.writer(responses_output)
+        responses_writer.writerow(['ID', 'Prompt ID', 'LLM', 'Response', 'Jailbroken', 'Note'])
+        for response in responses:
+            responses_writer.writerow([
+                response.get('id', ''),
+                response.get('prompt_id', ''),
+                response.get('llm', ''),
+                response.get('response', ''),
+                'Yes' if response.get('jailbroken', False) else 'No',
+                response.get('note', '') or ''
+            ])
+        responses_csv = responses_output.getvalue()
+        
+        # Return both CSVs as JSON (frontend will handle download)
+        return jsonify({
+            'success': True,
+            'data': {
+                'prompts': prompts_csv,
+                'responses': responses_csv,
+                'prompts_count': len(prompts),
+                'responses_count': len(responses)
+            }
+        }), 200
+    except Exception as e:
+        import traceback
+        print(f"Error in download_csv: {e}")
+        print(traceback.format_exc())
+        return jsonify({'error': str(e)}), 500
+
+
 if __name__ == '__main__':
     port = int(os.environ.get('PORT', 7860))
     app.run(host='0.0.0.0', port=port, debug=False)