app.py

from flask import Flask, request, render_template, jsonify, send_from_directory
import os
from notesync_core import process_youtube_link, process_media_file

# יצירת אפליקצייה
app = Flask(__name__, static_url_path='/static', static_folder='static')

# הגדרת תיקיית העלאות לקבצי אודיו
UPLOAD_FOLDER = "uploads"
os.makedirs(UPLOAD_FOLDER, exist_ok=True)  # יוצרת את התיקייה אם היא לא קיימת

# מסלול להחזרת קובץ שהועלה (עבור נגינה בדפדפן)
@app.route('/uploads/<path:filename>')
def serve_uploaded_file(filename):
    return send_from_directory(UPLOAD_FOLDER, filename)

# מסלול לדף הבית 
@app.route('/')
def home():
    return render_template("index.html")

# מסלול לעיבוד קובץ או קישור יוטיוב
@app.route('/process', methods=['POST'])
def process():
    # שליפת פרמטרים מהטופס
    mode = request.form.get("mode", "base").strip().lower()
    youtube_link = request.form.get("youtube_link", "").strip()
    media_file = request.files.get("media_file")

    try:
        # אם הוזן קישור ליוטיוב
        if youtube_link:
            print(f"📥 Received YouTube link: {youtube_link}")
            results = process_youtube_link(youtube_link, mode) 
            print("📤 JSON response (YouTube):", results)
            save_segments_text(results)  

        # אם הועלה קובץ מדיה
        elif media_file and media_file.filename != "":
            filename = media_file.filename
            media_path = os.path.join(UPLOAD_FOLDER, filename)

            print(f"📁 Received file: {filename}")
            media_file.save(media_path) 
            print(f"✅ Saved file at: {media_path}")

            results = process_media_file(media_path, mode) 
            print("📤 JSON response (File):", results)
            save_segments_text(results)  

        else:
            return jsonify({"error": "No input provided."}) 

    except Exception as e:
        print(f"❌ Error during processing: {e}") 
        return jsonify({"error": str(e)})

    return jsonify(results)  

# פונקציה ששומרת את התמלול כקובץ טקסט
def save_segments_text(data, folder="uploads"):
    os.makedirs(folder, exist_ok=True)
    audio_filename = data.get("audio_filename", "default_output").replace(".mp3", "")
    filename = f"{audio_filename}.txt"

    # איסוף כל המשפטים מכל הסגמנטים
    sentences = [sentence["text"] for segment in data.get("segments", []) for sentence in segment.get("sentences", [])]
    text_content = "\n".join(sentences)

    # שמירה לקובץ טקסט באותה תיקייה
    file_path = os.path.join(folder, filename)
    with open(file_path, "w", encoding="utf-8") as file:
        file.write(text_content)

# הרצת השרת בפיתוח
if __name__ == "__main__":
    app.run(debug=True)