|
from flask import Flask, render_template, request, jsonify |
|
import os |
|
import google.generativeai as genai |
|
import time |
|
|
|
app = Flask(__name__) |
|
|
|
|
|
token = os.environ.get("TOKEN") |
|
if not token: |
|
raise ValueError("La variable d'environnement TOKEN n'est pas définie.") |
|
|
|
genai.configure(api_key=token) |
|
|
|
generation_config = { |
|
"temperature": 1, |
|
"max_output_tokens": 8192, |
|
} |
|
|
|
safety_settings = [ |
|
{"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"}, |
|
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"}, |
|
{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"}, |
|
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"}, |
|
] |
|
|
|
model = genai.GenerativeModel( |
|
model_name="gemini-1.5-flash-002", |
|
generation_config=generation_config, |
|
safety_settings=safety_settings |
|
) |
|
|
|
source_languages = ["Francais", "English", "Spanish"] |
|
target_languages = ['nzebi', 'Fang(ntumu)','Fang(complet)','dikota', 'yipunu', 'omyene_nkomi'] |
|
|
|
lang_files = { |
|
'nzebi': 'Inzèbi.txt', |
|
'gisir': 'Gisir.txt', |
|
'dikota': 'dd.txt', |
|
'yipunu': 'yipunu.pdf', |
|
'Fang(complet)':'document_modifie.pdf', |
|
'akele': 'Akélé.txt', |
|
'Ghétsogo': 'Ghétsogo.txt', |
|
'Shimo': 'Shimo.txt', |
|
'omyene_nkomi': 'Omyènè_Nkomi.txt', |
|
'isangu': 'Isangu.txt', |
|
'Liwanzi': 'Liwanzi.txt', |
|
'Fang(ntumu)': 'Fang(ntumu).txt' |
|
} |
|
|
|
def main_fang(query): |
|
return "En maintenance" |
|
|
|
async def translate(input_text, source, target): |
|
if target == 'yipunu': |
|
try: |
|
pdf_path = lang_files[target] |
|
d = time.time() |
|
if not os.path.exists(pdf_path): |
|
return f"Erreur: Le fichier {pdf_path} n'existe pas." |
|
|
|
sample_file = genai.upload_file(path=pdf_path, display_name="yipunu-reference") |
|
s = time.time() |
|
print(s-d) |
|
|
|
prompt = f""" |
|
Using the provided Yipunu language reference PDF, translate the following text: |
|
Input text: {input_text} |
|
|
|
Please provide only the translation without any additional explanation. |
|
""" |
|
|
|
print(input_text) |
|
content = [sample_file, prompt] |
|
response = model.generate_content(content, request_options={"timeout": 600}) |
|
|
|
genai.delete_file(sample_file.name) |
|
print(response.text) |
|
return response.text |
|
|
|
except Exception as e: |
|
return f"Erreur lors de la traduction: {str(e)}" |
|
|
|
elif target == 'Fang(complet)': |
|
try: |
|
pdf_path = lang_files[target] |
|
d = time.time() |
|
if not os.path.exists(pdf_path): |
|
return f"Erreur: Le fichier {pdf_path} n'existe pas." |
|
|
|
sample_file = genai.upload_file(path=pdf_path, display_name="yipunu-reference") |
|
s = time.time() |
|
print(s-d) |
|
|
|
prompt = f""" |
|
Using the provided Fang language reference PDF, translate the following text: |
|
Input text: {input_text} |
|
|
|
Please provide only the translation without any additional explanation. |
|
""" |
|
|
|
print(input_text) |
|
content = [sample_file, prompt] |
|
response = model.generate_content(content, request_options={"timeout": 600}) |
|
|
|
genai.delete_file(sample_file.name) |
|
print(response.text) |
|
return response.text |
|
|
|
except Exception as e: |
|
return f"Erreur lors de la traduction: {str(e)}" |
|
|
|
else: |
|
chemin_fichier = lang_files[target] |
|
with open(chemin_fichier, 'r', encoding='utf-8') as fichier: |
|
contenu_langue_arrivee = fichier.read() |
|
|
|
tt = f""" |
|
contexte: {contenu_langue_arrivee} |
|
|
|
Utillisez les éléments de contexte suivants pour répondre à la question à la fin. |
|
Si vous ne connaissez pas la réponse, traduisez ce que vous pouvez et reecriver |
|
les autre comme ca, n'essayez pas d'inventer une réponse. Je veux que tu agisses |
|
comme un traducteur {target}. Je parle en français et tu traduis en {target} en |
|
te basant sur le contexte. Je ne veux aucune explication. Juste la réponse. |
|
Traduit ca < {input_text} > |
|
""" |
|
print(input_text) |
|
response = model.generate_content(tt) |
|
print(response.text) |
|
return response.text |
|
|
|
@app.route('/', methods=['GET', 'POST']) |
|
def index(): |
|
if request.method == 'POST': |
|
input_text = request.form['input_text'] |
|
source_language = request.form['source_language'] |
|
target_language = request.form['target_language'] |
|
translated_text = translate(input_text, source_language, target_language) |
|
return jsonify({'translated_text': translated_text}) |
|
return render_template('index.html', source_languages=source_languages, target_languages=target_languages) |
|
|
|
if __name__ == '__main__': |
|
app.run(debug=True) |