Upload 2 files

app.py

@@ -4,40 +4,11 @@ from helper_functions import *
 
 with gr.Blocks() as app:
     gr.Markdown('# FundedNext Customer Service Chatbot')
+    # Message Archived, Message Current, User Message, Bot Message
     session_data = gr.State([
-        [{"role": "system", "content": pre_text}],[]
+        [{"role": "system", "content": pre_text}],[],[],[]
     ])
-    def user(user_message, history):
-        return "", history + [[user_message, None]]
 
-    def bot(history, session_data_fn):
-        messages_archived = session_data_fn[0]
-        messages_current = session_data_fn[1]
-        bot_message, messages_archived, messages_current = get_reply(history[-1][0], messages_archived, messages_current)
-        history[-1][1] = bot_message
-        session_data_fn[0] = messages_archived
-        session_data_fn[1] = messages_current
-        return history, session_data_fn
-    
-    def reset_memory(session_data_fn):
-        messages_archived = session_data_fn[0]
-        if(len(messages_archived)>=21):
-            messages_archived = messages_archived[0:1] + messages_archived[3:]     
-            session_data_fn[0] = messages_archived
-        return session_data_fn
-
-    def clear_data(session_data_fn):
-        messages_archived = [
-        {"role": "system", "content": pre_text}
-        ]   
-        messages_current = []   
-        session_data_fn[0] = messages_archived
-        session_data_fn[1] = messages_current
-        return None, session_data_fn
-
-    def get_context_gr(session_data_fn):
-        messages_current = session_data_fn[1]
-        return str(messages_current)
 
     with gr.Tab("Chat"):
         with gr.Row():
@@ -52,6 +23,11 @@ with gr.Blocks() as app:
     with gr.Tab("Prompt"):
         context = gr.Textbox()
         submit_p = gr.Button("Check Prompt")
+    
+    with gr.Tab("Download"):
+        file = gr.File(interactive = False)
+        file_download_btn = gr.Button("Downlaod Chat History")
+
     # Tab Chat
     msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
         bot, [chatbot, session_data], [chatbot, session_data]
@@ -71,4 +47,7 @@ with gr.Blocks() as app:
     )
     # Tab Prompt
     submit_p.click(get_context_gr, session_data, context, queue=False)
+    # Tab Download
+    file_download_btn.click(fn = download_file, inputs = session_data, outputs = file, queue = False)
+# app.launch(debug=True)
 app.launch(auth=(os.getenv("id"), os.getenv("password")), show_api=False)

helper_functions.py

@@ -3,6 +3,7 @@ import os
 import openai
 import pandas as pd
 import numpy as np
+import time
 
 # Set up OpenAI API key
 openai.api_key = os.getenv("OPENAI_API_KEY")
@@ -37,11 +38,12 @@ def search(df, query, max_n, max_token):
 
 def get_context(query):
     results = search(df, query, max_n = 10, max_token = 500)
-    return f"""I will ask you questions based on the following context:
-— Start of Context —
+    return f"""I will ask you a question. You will answer with the help of knowlwedge base. 
+    Sometimes the knowledge base will be empty. It means that you will have to answer the question yourself.
+--Start of Knowledge Base--
 {results}
-— End of Context —
-My question is: “{query}”
+--End of Knowledge Base--
+My question is: "{query}"
 """
 
 def get_reply(message, messages_archived, messages_current):
@@ -65,3 +67,49 @@ def get_reply(message, messages_archived, messages_current):
 
     return reply, messages_archived, messages_current
 
+
+def user(user_message, history):
+    return "", history + [[user_message, None]]
+
+def bot(history, session_data_fn):
+    messages_archived = session_data_fn[0]
+    messages_current = session_data_fn[1]
+    bot_message, messages_archived, messages_current = get_reply(history[-1][0], messages_archived, messages_current)
+    history[-1][1] = bot_message
+    session_data_fn[0] = messages_archived
+    session_data_fn[1] = messages_current
+    session_data_fn[2].append(history[-1][0])
+    session_data_fn[3].append(history[-1][1])
+    return history, session_data_fn
+
+def reset_memory(session_data_fn):
+    messages_archived = session_data_fn[0]
+    # print("Message Archived Len=", len(messages_archived))
+    if(len(messages_archived)>=21):
+        messages_archived = messages_archived[0:1] + messages_archived[3:]     
+        session_data_fn[0] = messages_archived
+    return session_data_fn
+
+def clear_data(session_data_fn):
+    messages_archived = [
+    {"role": "system", "content": pre_text}
+    ]   
+    messages_current = []   
+    session_data_fn[0] = messages_archived
+    session_data_fn[1] = messages_current
+    session_data_fn[2] = []
+    session_data_fn[3] = []
+    return None, session_data_fn
+
+def get_context_gr(session_data_fn):
+    messages_current = session_data_fn[1]
+    return str(messages_current)
+
+def download_file(session_data_fn):
+    df = pd.DataFrame(list(zip(session_data_fn[2], session_data_fn[3])), 
+                      columns =['user_message', 'bot_reply'])
+    current_date_time = time.strftime("Date %Y_%m_%d Time %H_%M_%S", time.localtime())
+    file_name = "files/Chat History " + current_date_time + ".csv"
+    df.to_csv(file_name, index=False)
+    return file_name
+