Update app.py

app.py

@@ -1,9 +1,14 @@
 import streamlit as st
 import openai
-
+from googletrans import Translator
 # Set up OpenAI API credentials
 openai.api_key = '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'
 
+def translate_text(text, target_language):
+    translator = Translator()
+    translation = translator.translate(text, dest=target_language)
+    return translation.text
+
 def get_food_recommendation(prompt):
     # Generate a response using the ChatGPT API
     response = openai.Completion.create(
@@ -20,8 +25,17 @@ def get_food_recommendation(prompt):
 
     return recommendation
 
-# Streamlit app
-def main():
+def korean_text():
+    st.title("식품 및 영양 권장 시스템")
+    st.write("식품 및 영양 권장 시스템에 오신 것을 환영합니다")
+    st.markdown("웹페이지에서 도움이 될 수 있는 몇 가지 예시입니다:")
+    st.markdown("- 200칼로리 음식 한국어")
+    st.markdown("- 소고기 볶음밥 레시피")
+    st.markdown("- 쌀 새우 스파이스 레시피")
+    st.markdown("- 당뇨병에 좋은 음식 한국어")
+
+
+def eng_func():
     st.title("Food and Nutrition Recommendation System")
     st.write("Welcome to Food and Nutrition Recommendation System")
     st.markdown('''The following are some examples that the webpage can help with :
@@ -30,17 +44,37 @@ def main():
     - recipe for beef fried rice    
     - rice shrimp spices recipe   
     - foods for diabetes Korean''')
-    prompt = st.text_area("Enter your request for food recommendation")
 
-    if st.button("Get Recommendation"):
+# Streamlit app
+def main():
+    target_language = "ko"  # Set the target language to Korean
+
+    col1, col2 = st.columns(2)
+
+    with col1:
+        eng_func()
+
+    with col2:
+        korean_text()
+
+    prompt = st.text_area("Enter your request for food recommendation/ 음식 추천 요청을 입력하세요 ")
+
+    if st.button("Get Recommendation/추천 받기"):
         if prompt:
             recommendation_prompt = f"I'm looking for a food recommendation with the complete ingredients list based on the following criteria: {prompt}"
             recommendation = get_food_recommendation(recommendation_prompt)
-            st.markdown("Recommendation: " + recommendation)
+            translated_recommendation = translate_text(recommendation, target_language)
+
+            st.markdown("Recommendation (English / Korean):")
+            col3, col4 = st.columns(2)
+
+            with col3:
+                st.write(recommendation)
+
+            with col4:
+                st.write(translated_recommendation)
         else:
             st.warning("Please enter your request.")
 
 if __name__ == "__main__":
     main()
-
-