Update app.py

app.py

@@ -1,20 +1,10 @@
 import gradio as gr
-import pandas as pd
-from io import StringIO
 import openai
 import pytesseract
 import random
-
-
-
 import os
-
-import numpy as np
 import time
 
-# Set up OpenAI API key
-openai.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}
 
@@ -50,30 +40,29 @@ def create_data(files):
     return question_context
 
 
-########### TAB 2 (CHROMA + PLOT) FUNCTIONS  #############################
-
 ########### TAB 3 (CHAT) FUNCTIONS  #############################
 
 def user(user_message, history):
         return "", history + [[user_message, None]]
 
-def bot(history, messages_history, 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):
     user_message = history[-1][0]
 
-    bot_message, messages_history = ask_gpt(user_message, messages_history, 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)
     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, 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):
     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?'}]
-    print(messages_history)
+    
+    openai.api_key = api_key
     response = openai.ChatCompletion.create(
         model=chatgpt_model,
         messages=messages_history,
@@ -97,20 +86,16 @@ with gr.Blocks() as demo:
     
     
     ############# TAB 1 ##########################
-    with gr.Tab("Upload documents"):
+    with gr.Tab("Upload documents and create context"):
+        with gr.Row():
+            api_key = gr.Textbox(value='', type='password', label='Insert OPENAI API-key here')
         with gr.Row():
             files = gr.File(file_count='multiple', file_types=['image'], interactivate = True)
-            gr.Markdown("")
-            testi = gr.Button()
-        
-    ############# TAB 2 ##########################
-    with gr.Tab("Create questions"):
-        
+            create_context_btn = gr.Button(value='Recognize text and create context')
         with gr.Row():
             gr.Markdown("")
         with gr.Row():
-            create_context_btn = gr.Button()
-            teksti_contexti = gr.Textbox(value='Tähän tulee konteksti', label='Dataframe columns')
+            teksti_contexti = gr.Textbox(value='Tähän tulee konteksti', label='Created context')
     
     ############# TAB 3 ##########################
 
@@ -140,15 +125,12 @@ with gr.Blocks() as demo:
 
 
     # TAB 1 (UPLOAD) Interactive elements:
-    testi.click(print_files, [files])
-
-    # TAB 2 (CHROMA + PLOT) Interactive elements:
     create_context_btn.click(create_data, files, teksti_contexti)
 
 
     # 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, 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], [chatbot, state, msg_log]
     )
     clear.click(lambda: None, None, chatbot, queue=False).success(init_history, [state, system_prompt], [state, system_prompt, msg_log])