Update app.py

app.py

@@ -8,6 +8,21 @@ import time
 Init_system_prompt = "You are an AI Assistant that tries to teach kids various subjects. You are given learning material and you task is to ask questions given the material and then you also grade answers and give feedback how to improve the answers"
 system_message = {"role": "system", "content": Init_system_prompt}
 
+system_prompts = {
+    "English": "You are an AI Assistant that tries to teach kids various subjects. You are given learning material and you task is to ask questions given the material and then you also grade answers and give feedback how to improve the answers",
+    "Finnish": "Olet tekoälyavustaja jonka tehtävänä on auttaa lapsia oppimaan koulussa. Sinulle annetaan oppimateriaalia tekstinä ja sinun tehtäväsi on kysyä kysymyksiä annetusta tekstistä, arvostella vastauksia ja antaa palautetta kuinka vastauksia voidaan parantaa."
+}
+
+question_strings = {
+    "English": "\n Please ask a question about the previous paragramph: Question:",
+    "Finnish": "\n Kysy kysymys edellisen kappaleen perusteella. Kysymys:",
+}
+
+lang_mapping = {
+    "English": "en",
+    "Finnish": "fin"
+}
+
 import os
 from PIL import Image
 
@@ -27,17 +42,18 @@ def print_files(files):
         print(file.file)
 
 
-def create_data(files):
+def create_data(files, language_selection):
     question_context = ''
     for file in files:
         if file.name.endswith('png') or file.name.endswith('.jpg'):
             try:
-                question_context += (pytesseract.image_to_string(Image.open(file.name))) + '\n\n'
+                question_context += (pytesseract.image_to_string(Image.open(file.name), lang=lang_mapping[language_selection])) + '\n\n'
             except Exception as e:
                 print(e)
                 pass
+    system_prompt = system_prompts[language_selection]
 
-    return question_context
+    return question_context, system_prompt
 
 
 ########### TAB 3 (CHAT) FUNCTIONS  #############################
@@ -45,22 +61,22 @@ def create_data(files):
 def user(user_message, history):
         return "", history + [[user_message, None]]
 
-def bot(history, messages_history, api_key, system_prompt, teksti_contexti, temperature, max_tokens, chatgpt_model, max_context_size_for_question):
+def bot(history, messages_history, api_key, system_prompt, teksti_contexti, temperature, max_tokens, chatgpt_model, max_context_size_for_question, language_selection):
     user_message = history[-1][0]
 
-    bot_message, messages_history = ask_gpt(user_message, messages_history, api_key, system_prompt, teksti_contexti, temperature, max_tokens, chatgpt_model, max_context_size_for_question)
+    bot_message, messages_history = ask_gpt(user_message, messages_history, api_key, system_prompt, teksti_contexti, temperature, max_tokens, chatgpt_model, max_context_size_for_question, language_selection)
     messages_history += [{"role": "assistant", "content": bot_message}]
     history[-1][1] = bot_message
     time.sleep(0.2)
     return history, messages_history, str(messages_history)
 
-def ask_gpt(message, messages_history, api_key, system_prompt, context, temperature, max_tokens, chatgpt_model, max_context_size_for_question):
+def ask_gpt(message, messages_history, api_key, system_prompt, context, temperature, max_tokens, chatgpt_model, max_context_size_for_question, language_selection):
     messages_history, system_prompt, _ = init_history(messages_history, system_prompt)
     if len(messages_history) < 1:
         messages_history = [{"role": "system", "content": system_prompt}]
     max_possible_position = len(context)- max_context_size_for_question
     start = random.randint(0,max_possible_position)
-    messages_history += [{"role": "user", "content": context[start:start+max_context_size_for_question] + '\n Please ask a question about the previous paragramph?'}]
+    messages_history += [{"role": "user", "content": context[start:start+max_context_size_for_question] + question_strings[language_selection]}]
     
     openai.api_key = api_key
     response = openai.ChatCompletion.create(
@@ -90,6 +106,7 @@ with gr.Blocks() as demo:
         with gr.Row():
             api_key = gr.Textbox(value='', type='password', label='Insert OPENAI API-key here')
         with gr.Row():
+            language_selection = gr.Dropdown(value='English', choices=["English","Finnish"], label='Select language')
             files = gr.File(file_count='multiple', file_types=['image'], interactivate = True)
             create_context_btn = gr.Button(value='Recognize text and create context')
         with gr.Row():
@@ -125,12 +142,12 @@ with gr.Blocks() as demo:
 
 
     # TAB 1 (UPLOAD) Interactive elements:
-    create_context_btn.click(create_data, files, teksti_contexti)
+    create_context_btn.click(create_data, [files, language_selection], [teksti_contexti, system_prompt])
 
 
     # TAB 3 (CHAT) Interactive elements:
     msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, [chatbot, state, api_key, system_prompt, teksti_contexti, temperature, max_tokens, chatgpt_model, max_context_size_for_question], [chatbot, state, msg_log]
+        bot, [chatbot, state, api_key, system_prompt, teksti_contexti, temperature, max_tokens, chatgpt_model, max_context_size_for_question, language_selection], [chatbot, state, msg_log]
     )
     clear.click(lambda: None, None, chatbot, queue=False).success(init_history, [state, system_prompt], [state, system_prompt, msg_log])