Update app.py

Updated app.py

app.py

@@ -1,25 +1,18 @@
-from flask import Flask, request, jsonify
+import streamlit as st
 from transformers import pipeline
 
-# Load translation model
+# Load the translation model
 translator = pipeline('translation_en_to_uk', model='Helsinki-NLP/opus-mt-en-uk')
 
-# Create Flask app
-app = Flask(__name__)
+# Streamlit UI
+st.title("English to Ukrainian Translator")
 
-@app.route("/")
-def home():
-    return open("index.html").read()
+text = st.text_area("Enter English text:")
 
-@app.route("/translate", methods=["POST"])
-def translate():
-    data = request.json
-    text = data.get("text", "")
-    if not text:
-        return jsonify({"error": "No text provided"}), 400
-
-    translated_text = translator(text, max_length=100)[0]['translation_text']
-    return jsonify({"translation": translated_text})
-
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=8080)
+if st.button("Translate"):
+    if text:
+        translated_text = translator(text, max_length=100)[0]['translation_text']
+        st.subheader("Translation:")
+        st.write(translated_text)
+    else:
+        st.warning("Please enter text to translate.")