Spaces:
Running
Running
| 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", use_fast=False) | |
| 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}" | |
| 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) | |