app.py

import streamlit as st
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
#from lyricsgenius import Genius
import os

# Wybór modelu do tłumaczenia
#model_name = "clui/opus-it-pl-v1"  # Model tłumaczenia z włoskiego na polski
model_name = "clui/fine3"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)

# Tytuł aplikacji
st.title("Neural Notes")

# Ustawienia tła
st.markdown(
    """
    <style>
    body {
        background-color: #f0f4f8; /* Kolor tła podobny do DeepL */
        color: #333333; /* Kolor tekstu */
    }
    .stButton>button {
        background-color: #4CAF50; /* Zielony przycisk */
        color: white;
    }
    </style>
    """,
    unsafe_allow_html=True
)

# Sekcja Translator
st.subheader("Translator")
st.write("Wprowadź tekst piosenki, a model przetłumaczy go linijka po linijce. Umożliwia to łatwe zrozumienie treści utworów w innym języku.")

# Wprowadzenie tekstu piosenki
input_text = st.text_area("Wprowadź tekst, model przetłumaczy go linijka po linijce:", height=300)

if st.button("Tłumacz"):
    if input_text:
        # Podział tekstu na linie
        lines = input_text.split('\n')
        translated_lines = []
        
        # Tłumaczenie każdej linii
        for line in lines:
            if line.strip():  # Sprawdzenie, czy linia nie jest pusta
                # Tokenizacja
                inputs = tokenizer(line, return_tensors="pt", padding=True)
                
                # Generowanie tłumaczenia
                translated_outputs = model.generate(**inputs)
                
                # Dekodowanie przetłumaczonego tekstu
                translated_text = tokenizer.decode(translated_outputs[0], skip_special_tokens=True)
                
                translated_lines.append(translated_text)
            else:
                translated_lines.append("")  # Dodanie pustej linii jako odstępu

        # Wyświetlanie przetłumaczonych linii w dwóch kolumnach
        col1, col2 = st.columns(2)  # Dwie kolumny
        
        with col1:
            st.subheader("Oryginalne linie:")
            for line in lines:
                st.write(line)

        with col2:
            st.subheader("Przetłumaczone linie:")
            for line in translated_lines:
                st.write(line)
    else:
        st.warning("Proszę wpisać tekst piosenki.")