Update app.py

app.py

@@ -4,6 +4,7 @@ import pickle
 import streamlit as st
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 
+from translator import translate
 
 def init_session_state():
     if 'history' not in st.session_state:
@@ -55,10 +56,27 @@ if text:
 
     answer = dictionary[output][output2]
 
+    # Create a translator object with specified source and target languages
+    translator = Translator(from_lang='el', to_lang='en')
+    translator2 = Translator(from_lang='en', to_lang='el')
+
+ 
+
+# Translate the text from Greek to English
+    answer = translator.translate(answer)
+    text = translator.translate(text)
+
+# text_to_translate2 = text[499:999]
+# translated_text2 = translator.translate(text_to_translate2)
+
+
 
     st.session_state.history += "Based on the following information" + answer +"answer this question:" + text  # Add new text to history
     out = pipe(st.session_state.history)  # Generate output based on history
-    st.text(out[0]['generated_text'])
+    
+    translated_text2 = translator2.translate(out[0]['generated_text'])
+
+    st.text(translated_text2)
     # st.text("History: " + st.session_state.history)
 
     # st.text(output)