Enhance app with file upload, error handling, and localization

.gitattributes

@@ -1 +1,6 @@
 *.db filter=lfs diff=lfs merge=lfs -text
+*.edf filter=lfs diff=lfs merge=lfs -text
+*.bdf filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text

app.py

@@ -4,24 +4,42 @@ import sqlite3
 import requests
 import random
 import time
+import traceback
 from flask import Flask, render_template, request, jsonify, g
 from flask_cors import CORS
+from werkzeug.utils import secure_filename
 
 app = Flask(__name__)
 CORS(app)
 
 # Configuration
 app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 16MB
+app.config['UPLOAD_FOLDER'] = os.path.join(app.instance_path, 'uploads')
 app.config['DATABASE'] = os.path.join(app.instance_path, 'synapse.db')
 SILICONFLOW_API_KEY = "sk-vimuseiptfbomzegyuvmebjzooncsqbyjtlddrfodzcdskgi"
 SILICONFLOW_API_URL = "https://api.siliconflow.cn/v1/chat/completions"
 
-# Ensure instance folder exists
+# Ensure instance and upload folders exist
 try:
     os.makedirs(app.instance_path)
+    os.makedirs(app.config['UPLOAD_FOLDER'])
 except OSError:
     pass
 
+# Error Handlers
+@app.errorhandler(404)
+def page_not_found(e):
+    return render_template('index.html'), 404
+
+@app.errorhandler(500)
+def internal_server_error(e):
+    traceback.print_exc()
+    return jsonify(error="Internal Server Error", message=str(e)), 500
+
+@app.errorhandler(413)
+def request_entity_too_large(e):
+    return jsonify(error="File too large", message="File exceeds the maximum allowed size of 16MB"), 413
+
 # Database Setup
 def get_db():
     db = getattr(g, '_database', None)
@@ -49,16 +67,66 @@ def init_db():
                 raw_metrics TEXT
             )
         ''')
+        
+        # Check if empty and add default data
+        cur = db.execute('SELECT count(*) FROM history')
+        if cur.fetchone()[0] == 0:
+            default_analysis = {
+                "summary": "**演示数据报告**: 这是一个自动生成的示例报告。\n\n受试者 **DEMO-USER** 在 **专注训练** 中表现优秀。Alpha波与Beta波的交替出现表明受试者能够自如地在放松和专注状态间切换。",
+                "cognitive_state": "Flow State (心流状态)",
+                "recommendations": ["继续保持当前的训练频率", "尝试增加训练时长至30分钟", "记录训练后的主观感受"],
+                "radar_chart": {"专注": 85, "放松": 70, "反应速度": 80, "记忆负荷": 65, "情绪稳定性": 88}
+            }
+            db.execute(
+                'INSERT INTO history (timestamp, subject_id, session_type, analysis_result, raw_metrics) VALUES (?, ?, ?, ?, ?)',
+                (time.strftime('%Y-%m-%d %H:%M:%S'), 'DEMO-USER', 'Focus Training', json.dumps(default_analysis), "demo_data")
+            )
+            print("Default data initialized.")
+            
         db.commit()
 
 init_db()
 
+# --- Helpers ---
+ALLOWED_EXTENSIONS = {'txt', 'csv', 'json', 'edf', 'bdf'}
+
+def allowed_file(filename):
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
 # --- Routes ---
 
 @app.route('/')
 def index():
     return render_template('index.html')
 
+@app.route('/api/upload', methods=['POST'])
+def upload_file():
+    if 'file' not in request.files:
+        return jsonify({'error': 'No file part'}), 400
+    file = request.files['file']
+    if file.filename == '':
+        return jsonify({'error': 'No selected file'}), 400
+    if file and allowed_file(file.filename):
+        filename = secure_filename(file.filename)
+        filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+        file.save(filepath)
+        
+        # Mock processing of the file to extract metrics
+        # In a real app, we would parse EDF/CSV here
+        file_size = os.path.getsize(filepath)
+        
+        return jsonify({
+            'status': 'success',
+            'message': f'File {filename} uploaded successfully ({file_size} bytes)',
+            'extracted_metrics': {
+                'alpha': random.uniform(8, 12),
+                'beta': random.uniform(15, 25),
+                'theta': random.uniform(4, 7),
+                'delta': random.uniform(1, 3)
+            }
+        })
+    return jsonify({'error': 'File type not allowed'}), 400
+
 @app.route('/api/mock/signal')
 def mock_signal():
     """Generate mock EEG signal data for visualization."""

templates/index.html

@@ -152,6 +152,19 @@
                                         <option>Motor Imagery (运动想象)</option>
                                     </select>
                                 </div>
+                                
+                                <!-- File Upload Section -->
+                                <div>
+                                    <label class="block text-sm font-medium text-slate-700 mb-1">数据源 (可选)</label>
+                                    <div class="flex gap-2">
+                                        <button @click="triggerUpload" class="flex-1 px-3 py-2 border border-dashed border-slate-300 rounded-lg text-slate-500 hover:bg-slate-50 hover:text-brand-600 transition-colors text-sm flex items-center justify-center gap-2">
+                                            <i class="fa-solid fa-cloud-arrow-up"></i>
+                                            ${ uploadStatus || '上传 EEG/CSV 文件' }
+                                        </button>
+                                        <input type="file" ref="fileInput" @change="handleFileUpload" class="hidden" accept=".csv,.txt,.edf,.bdf,.json">
+                                    </div>
+                                </div>
+
                                 <button @click="analyzeSession" :disabled="isAnalyzing" class="w-full mt-auto bg-brand-600 hover:bg-brand-700 text-white py-2.5 rounded-lg font-medium transition-all shadow-lg shadow-brand-500/30 disabled:opacity-50 disabled:cursor-not-allowed flex justify-center items-center gap-2">
                                     <i v-if="isAnalyzing" class="fa-solid fa-circle-notch animate-spin"></i>
                                     ${ isAnalyzing ? '正在分析数据...' : '生成分析报告' }
@@ -283,6 +296,8 @@
                 const metrics = ref({ alpha: 0, beta: 0, theta: 0, delta: 0 });
                 const lastAnalysis = ref(null);
                 const historyList = ref([]);
+                const uploadStatus = ref('');
+                const fileInput = ref(null);
                 
                 // Chart Refs
                 const signalChart = ref(null);
@@ -304,6 +319,58 @@
                     return marked.parse(lastAnalysis.value.summary);
                 });
 
+                const triggerUpload = () => {
+                    fileInput.value.click();
+                };
+
+                const handleFileUpload = async (event) => {
+                    const file = event.target.files[0];
+                    if (!file) return;
+
+                    uploadStatus.value = '上传中...';
+                    const formData = new FormData();
+                    formData.append('file', file);
+
+                    try {
+                        const res = await fetch('/api/upload', {
+                            method: 'POST',
+                            body: formData
+                        });
+                        const data = await res.json();
+                        
+                        if (res.ok) {
+                            uploadStatus.value = '上传成功';
+                            // Use extracted metrics to update dashboard
+                            if (data.extracted_metrics) {
+                                metrics.value = data.extracted_metrics;
+                                // Also update spectrum chart
+                                if (spectrumChartInst) {
+                                    spectrumChartInst.setOption({
+                                        series: [{
+                                            data: [
+                                                data.extracted_metrics.alpha,
+                                                data.extracted_metrics.beta,
+                                                data.extracted_metrics.theta,
+                                                data.extracted_metrics.delta
+                                            ]
+                                        }]
+                                    });
+                                }
+                            }
+                            setTimeout(() => { uploadStatus.value = ''; }, 3000);
+                        } else {
+                            uploadStatus.value = '上传失败';
+                            alert(data.error || '上传失败');
+                        }
+                    } catch (e) {
+                        console.error(e);
+                        uploadStatus.value = '出错';
+                        alert('上传过程中发生错误');
+                    }
+                    // Reset input
+                    event.target.value = '';
+                };
+
                 const initCharts = () => {
                     if (signalChart.value) {
                         signalChartInst = echarts.init(signalChart.value);
@@ -490,8 +557,9 @@
                     currentView, mobileMenuOpen, isRecording, isAnalyzing,
                     subjectId, sessionType, metrics, lastAnalysis, historyList,
                     signalChart, spectrumChart, radarChart,
-                    parsedSummary,
-                    toggleRecording, analyzeSession, fetchHistory, loadAnalysis
+                    parsedSummary, uploadStatus, fileInput,
+                    toggleRecording, analyzeSession, fetchHistory, loadAnalysis,
+                    triggerUpload, handleFileUpload
                 };
             }
         }).mount('#app');