Noveramaaz commited on
Commit
46f80b5
·
verified ·
1 Parent(s): 4cb932e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -11
app.py CHANGED
@@ -1,8 +1,10 @@
1
- import gradio as gr
2
  from transformers import MarianMTModel, MarianTokenizer
3
  from langdetect import detect
4
  from sacremoses import MosesTokenizer, MosesDetokenizer
5
 
 
 
6
  # Supported languages for translation
7
  SUPPORTED_LANGUAGES = {
8
  "en": "English",
@@ -12,6 +14,18 @@ SUPPORTED_LANGUAGES = {
12
  "it": "Italian"
13
  }
14
 
 
 
 
 
 
 
 
 
 
 
 
 
15
  def translate_text(text, target_lang="fr"):
16
  # Detect input text language
17
  input_lang = detect(text)
@@ -32,14 +46,6 @@ def translate_text(text, target_lang="fr"):
32
 
33
  return translated_text
34
 
35
- # Function for translating text using Gradio
36
- def translate_interface(text, target_language):
37
- translated_text = translate_text(text, target_language)
38
- return translated_text
39
-
40
- # Create Gradio interface
41
- inputs = gr.Textbox(lines=3, label="Enter the text to translate")
42
- target_language = gr.Dropdown(choices=list(SUPPORTED_LANGUAGES.keys()), label="Target Language")
43
- output = gr.Textbox(placeholder="Translated Text", label="Translated Text")
44
 
45
- gr.Interface(fn=translate_interface, inputs=[inputs, target_language], outputs=output, title="Text Translation", theme="compact").launch(debug = True)
 
1
+ from flask import Flask, render_template, request, jsonify
2
  from transformers import MarianMTModel, MarianTokenizer
3
  from langdetect import detect
4
  from sacremoses import MosesTokenizer, MosesDetokenizer
5
 
6
+ app = Flask(__name__)
7
+
8
  # Supported languages for translation
9
  SUPPORTED_LANGUAGES = {
10
  "en": "English",
 
14
  "it": "Italian"
15
  }
16
 
17
+ @app.route("/")
18
+ def index():
19
+ return render_template("index.html")
20
+
21
+ @app.route("/translate", methods=["POST"])
22
+ def translate_text():
23
+ data = request.get_json()
24
+ text = data["text"]
25
+ target_lang = data["target_language"]
26
+ translated_text = translate_text(text, target_lang)
27
+ return jsonify({"translated_text": translated_text})
28
+
29
  def translate_text(text, target_lang="fr"):
30
  # Detect input text language
31
  input_lang = detect(text)
 
46
 
47
  return translated_text
48
 
49
+ if __name__ == "__main__":
50
+ app.run(debug=True)
 
 
 
 
 
 
 
51