Spaces:
Running
Running
File size: 3,078 Bytes
7a7248b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
from flask import Flask, render_template, request
import os
import subprocess
from speech_recognition import Recognizer, AudioFile
from werkzeug.utils import secure_filename
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
from deepmultilingualpunctuation import PunctuationModel
app = Flask(__name__)
yuklenen_dosyalar = "uploads"
app.config["yuklenen_dosyalar"] = yuklenen_dosyalar
dosya_turleri = {"mp3", "m4a"}
# Modelleri yükle
tokenizer = AutoTokenizer.from_pretrained("csebuetnlp/mT5_multilingual_XLSum")
model = AutoModelForSeq2SeqLM.from_pretrained("csebuetnlp/mT5_multilingual_XLSum")
punct_model = PunctuationModel(model="oliverguhr/fullstop-punctuation-multilingual-sonar-base")
def metin_duzenleme(text):
text = text.strip()
try:
return punct_model.restore_punctuation(text)
except Exception as e:
print("Noktalama hatası:", e)
return text
def metin_ozetleme(text):
girilen_metin = "tr: " + text
girilenler = tokenizer([girilen_metin], return_tensors="pt", max_length=10000, truncation=True)
summary_ids = model.generate(
girilenler["input_ids"],
max_length=1000,
min_length=1,
length_penalty=1.0,
num_beams=4,
early_stopping=True
)
return tokenizer.decode(summary_ids[0], skip_special_tokens=True)
def uygun_turler(filename):
return "." in filename and filename.rsplit(".", 1)[1].lower() in dosya_turleri
def convert_mp3_to_wav(mp3_path):
wav_path = mp3_path.rsplit('.', 1)[0] + '.wav'
subprocess.run(['ffmpeg', '-i', mp3_path, wav_path, '-y'], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
return wav_path
def ses_donusturucu(wav_path):
recognizer = Recognizer()
with AudioFile(wav_path) as source:
ses = recognizer.record(source)
try:
return recognizer.recognize_google(ses, language="tr-TR")
except Exception as e:
return f"Ses tanıma hatası: {e}"
@app.route("/", methods=["GET", "POST"])
def index():
if request.method == "POST":
if "file" not in request.files:
return "Dosya bulunamadı."
file = request.files["file"]
if file.filename == "":
return "Dosya seçilmedi."
if file and uygun_turler(file.filename):
filename = secure_filename(file.filename)
file_path = os.path.join(app.config["yuklenen_dosyalar"], filename)
file.save(file_path)
wav_path = convert_mp3_to_wav(file_path)
raw_text = ses_donusturucu(wav_path)
metin = metin_duzenleme(raw_text)
ozet = metin_ozetleme(metin)
return render_template("index.html", metin=metin, ozet=ozet)
return render_template("index.html")
if __name__ == "__main__":
if not os.path.exists(yuklenen_dosyalar):
os.makedirs(yuklenen_dosyalar)
port = int(os.environ.get("PORT", 7860)) # Hugging Face için 7860 port
app.run(host="0.0.0.0", port=port, debug=False)
|