Added Application files

.gitignore

@@ -0,0 +1 @@
+.env

dockerfile

@@ -0,0 +1,8 @@
+FROM python:3.11-slim
+WORKDIR /app
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+EXPOSE 5000
+CMD [ "python","language_detection.py" ]

language_detection.py

@@ -0,0 +1,105 @@
+import whisper
+import numpy as np
+from datetime import datetime
+from flask import Flask,request,jsonify
+from transformers import pipeline
+import os
+from dotenv import load_dotenv
+load_dotenv()
+app=Flask(__name__)
+model=whisper.load_model("base")
+SAMPLE_RATE=16000
+DURATION=10
+OUTPUT_FILE="recorded_audio.wav"
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-en",token=os.environ.get("HUGGING_FACE_AUTH_TOKEN"))
+
+
+def detect(audio_path):
+    audio=whisper.load_audio(audio_path)
+    audio=whisper.pad_or_trim(audio)
+
+    mel=whisper.log_mel_spectrogram(audio).to(model.device)
+    _,probs=model.detect_language(mel)
+
+    detected_lang=max(probs,key=probs.get)
+    conf=probs[detected_lang]
+
+    res=model.transcribe(audio_path,language=detected_lang)
+    trans=res["text"]
+
+    translation = trans if detected_lang == "en" else translator(trans)[0]["translation_text"]
+    return detected_lang,conf,trans,translation
+
+@app.route('/detect-language',methods=['POST'])
+def detect_endpoint():
+    try:
+        if 'audio' not in request.files:
+            return jsonify({"status":"error","message":"No audio file provided"}),400
+        else:
+            audio_file=request.files['audio']
+            audio_path=os.path.join("uploads",audio_file.filename)
+            os.makedirs("uploads",exist_ok=True)
+            audio_file.save(audio_path)
+        '''else:
+            audio_path=record()'''
+        det_lang,conf,_,_=detect(audio_path)
+        if os.path.exists(audio_path) and audio_path!=OUTPUT_FILE:
+            os.remove(audio_path)
+        return jsonify({
+            "status":"success",
+            "detected_language":det_lang,
+            "confidence":f"{conf:.2%}"
+        })
+    except Exception as e:
+        return jsonify({"status":"error","message":str(e)}),500
+    
+
+@app.route('/transcription',methods=['POST'])
+def transcription_endpoint():
+    try:
+        if 'audio' not in request.files:
+            return jsonify({"status":"error","message":"No audio file provided"}),400
+        else:
+            audio_file=request.files['audio']
+            audio_path=os.path.join("uploads",audio_file.filename)
+            os.makedirs("uploads",exist_ok=True)
+            audio_file.save(audio_path)
+        '''else:
+            audio_path=record()'''
+        _,_,trans,_=detect(audio_path)
+        if os.path.exists(audio_path) and audio_path!=OUTPUT_FILE:
+            os.remove(audio_path)
+        return jsonify({
+            "status":"success",
+            "transcription":trans
+        })
+    except Exception as e:
+        return jsonify({"status":"error","message":str(e)}),500
+    
+
+@app.route('/translation',methods=['POST'])
+def translation_endpoint():
+    try:
+        if 'audio' not in request.files:
+            return jsonify({"status":"error","message":"No audio file provided"}),400
+        else:
+            audio_file=request.files['audio']
+            audio_path=os.path.join("uploads",audio_file.filename)
+            os.makedirs("uploads",exist_ok=True)
+            audio_file.save(audio_path)
+        '''else:
+            audio_path=record()'''
+        det_lang,_,_,trans=detect(audio_path)
+        if os.path.exists(audio_path) and audio_path!=OUTPUT_FILE:
+            os.remove(audio_path)
+        return jsonify({
+            "status":"success",
+            "detected_language":det_lang,
+            "translation":trans
+        })
+    except Exception as e:
+        return jsonify({"status":"error","message":str(e)}),500
+
+if __name__=="__main__":
+    app.run(debug=True,host="0.0.0.0",port=5000)
+