Update app.py

app.py

@@ -8,6 +8,8 @@ from faster_whisper import WhisperModel
 import subprocess
 import shutil
 import datetime
+import json
+import csv
 
 # === Clean /tmp at startup ===
 TMP_DIR = "/tmp"
@@ -46,8 +48,19 @@ LANG_CODES = {
 model = WhisperModel("base", compute_type="int8")
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-# === In-memory session history ===
-session_history = []
+# === Persistent session history ===
+HISTORY_FILE = "history.json"
+def load_history():
+    if os.path.exists(HISTORY_FILE):
+        with open(HISTORY_FILE, "r", encoding="utf-8") as f:
+            return json.load(f)
+    return []
+
+def save_history(history):
+    with open(HISTORY_FILE, "w", encoding="utf-8") as f:
+        json.dump(history, f, ensure_ascii=False, indent=2)
+
+session_history = load_history()
 
 # === Audio Processing ===
 def convert_to_wav(input_file):
@@ -121,16 +134,30 @@ def tutor_feedback(audio_file, language):
     mp3_path = f"/tmp/{uuid.uuid4()}.mp3"
     tts.save(mp3_path)
 
-    # Store session history
     timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M")
-    session_history.append({
+    session_entry = {
         "timestamp": timestamp,
         "transcript": transcript,
         "feedback": feedback_text
-    })
+    }
+    session_history.append(session_entry)
+    save_history(session_history)
 
     return transcript, feedback_text, mp3_path, transcript
 
+# === Export history ===
+def export_history_json():
+    return json.dumps(session_history, indent=2, ensure_ascii=False)
+
+def export_history_csv():
+    csv_path = "/tmp/session_history.csv"
+    with open(csv_path, "w", newline='', encoding="utf-8") as f:
+        writer = csv.writer(f)
+        writer.writerow(["Timestamp", "Transcript", "Feedback"])
+        for s in session_history:
+            writer.writerow([s['timestamp'], s['transcript'], s['feedback']])
+    return csv_path
+
 # === Gradio Interface ===
 with gr.Blocks(css="light_mode_chatter_owl.css") as app:
     gr.Markdown(
@@ -163,15 +190,17 @@ with gr.Blocks(css="light_mode_chatter_owl.css") as app:
 
     example_box = gr.Textbox(label="🗣 Suggested Improvement", visible=True, placeholder="Click to generate improved speech...")
 
-    # Session History Display
     def get_session_table():
         rows = [[s['timestamp'], s['transcript'][:60] + '...', s['feedback'][:60] + '...'] for s in session_history]
         return rows
 
     history_display = gr.Dataframe(headers=["🕒 Timestamp", "📝 Transcript", "📋 Feedback Preview"], interactive=False)
     history_btn = gr.Button("🕓 View My Past Sessions")
+    export_json_btn = gr.Button("📤 Export History (JSON)")
+    export_csv_btn = gr.Button("📄 Export History (CSV)")
+    export_json_output = gr.Textbox(visible=False)
+    export_csv_output = gr.File(label="Download CSV")
 
-    # Connect functions
     audio_input.change(
         fn=tutor_feedback,
         inputs=[audio_input, language_dropdown],
@@ -196,7 +225,19 @@ with gr.Blocks(css="light_mode_chatter_owl.css") as app:
         outputs=history_display
     )
 
+    export_json_btn.click(
+        fn=export_history_json,
+        inputs=None,
+        outputs=export_json_output
+    )
+
+    export_csv_btn.click(
+        fn=export_history_csv,
+        inputs=None,
+        outputs=export_csv_output
+    )
+
 # === Launch App ===
 if __name__ == "__main__":
     print("✅ App is launching...")
-    app.launch(server_name="0.0.0.0", server_port=7860, debug=True)
+    app.launch(server_name="0.0.0.0", server_port=7860, debug=True)