Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 from io import StringIO
 import openai
 import pytesseract
+import random
 
 
 
@@ -12,7 +13,8 @@ import numpy as np
 import time
 
 # Set up OpenAI API key
-openai.api_key = os.environ["CHATGPT_API_KEY"]
+from api_key import api_key
+openai.api_key = api_key#os.environ["CHATGPT_API_KEY"]
 
 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}
@@ -26,7 +28,7 @@ path = os.system("which tesseract >> path.txt")
 with open("path.txt", 'r') as file:
     tesseract_path = file.read().replace('\n', '')
 
-print(tesseract_path)
+
 ########### TAB 1 (UPLOAD) FUNCTIONS  #############################
 
 def print_files(files):
@@ -56,7 +58,7 @@ def create_data(files):
 def user(user_message, history):
         return "", history + [[user_message, None]]
 
-def bot(history, messages_history, system_prompt, teksti_contexti, temperature, max_tokens, chatgpt_model):
+def bot(history, messages_history, system_prompt, teksti_contexti, temperature, max_tokens, chatgpt_model, max_context_size_for_question):
     user_message = history[-1][0]
 
     bot_message, messages_history = ask_gpt(user_message, messages_history, system_prompt, teksti_contexti, temperature, max_tokens,chatgpt_model)
@@ -65,10 +67,13 @@ def bot(history, messages_history, system_prompt, teksti_contexti, temperature,
     time.sleep(0.2)
     return history, messages_history, str(messages_history)
 
-def ask_gpt(message, messages_history, context, system_prompt, temperature, max_tokens, chatgpt_model):
+def ask_gpt(message, messages_history, system_prompt, context, temperature, max_tokens, chatgpt_model, max_context_size_for_question):
+    messages_history, system_prompt, _ = init_history(messages_history, system_prompt)
     if len(messages_history) < 1:
         messages_history = [{"role": "system", "content": system_prompt}]
-    messages_history += [{"role": "user", "content": context + '\n Please ask a question about the previous paragramph?'}]
+    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?'}]
     print(messages_history)
     response = openai.ChatCompletion.create(
         model=chatgpt_model,
@@ -117,8 +122,9 @@ with gr.Blocks() as demo:
         with gr.Row():
             system_prompt = gr.Textbox(value=Init_system_prompt, label='Insert system message here')
             chatgpt_model = gr.Dropdown(choices=["gpt-3.5-turbo", "gpt-3.5-turbo-0301", "gpt-3.5-turbo-0613"], value='gpt-3.5-turbo',label='ChatGPT model to use', interactive=True)
-            temperature = gr.Slider(minimum=0.0, maximum=1.0, step=0.05, value=0.0, label='temperature')
+            temperature = gr.Slider(minimum=0.0, maximum=1.0, step=0.05, value=0.7, label='temperature')
             max_tokens = gr.Slider(minimum=10, maximum=600, step=10, value=100, label='Max tokens')
+            max_context_size_for_question = gr.Slider(minimum=10, maximum=600, step=10, value=100, label='Max context for questions')
         with gr.Row():
             chatbot = gr.Chatbot(label='ChatGPT Chat')
             state = gr.State([])
@@ -143,7 +149,7 @@ with gr.Blocks() as demo:
 
     # TAB 3 (CHAT) Interactive elements:
     msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, [chatbot, state, system_prompt, teksti_contexti, temperature, max_tokens, chatgpt_model], [chatbot, state, msg_log]
+        bot, [chatbot, state, system_prompt, teksti_contexti, temperature, max_tokens, chatgpt_model, max_context_size_for_question], [chatbot, state, msg_log]
     )
     clear.click(lambda: None, None, chatbot, queue=False).success(init_history, [state, system_prompt], [state, system_prompt, msg_log])