Update app.py

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 import random
 import time
 import json
+import ast
 
 def get_completion(prompt, model="gpt-3.5-turbo"):
     messages = [{"role": "user", "content": prompt}]
@@ -47,11 +48,10 @@ DIVAR_CONTEXT = """
 """
 
 base_context = [ {'role':'system', 'content': DIVAR_CONTEXT} ]  # accumulate messages
-save = ""
-context = base_context.copy()
 
-def respond(message, chat_history, api_key):
-    global context, save
+
+def respond(message, chat_history, api_key, context_box):
+    context = ast.literal_eval(context_box)
     try:
         openai.api_key = api_key
         save = api_key
@@ -60,19 +60,17 @@ def respond(message, chat_history, api_key):
         
         messages = context.copy()
         messages.append(
-            {'role':'system', 'content':'create a json based on the previous conversation (Make sure to only output the json and nothing more). The fields should be 1) search_query (make it a list of possible queries which can find the best items for the user in Divar website) 2) price_range with min and max number In Tomans 3) possible_filters (keys must be in english)' },    
+            {'role':'system', 'content':'create a short json based on the previous conversation (Make sure to only output the json and nothing more). The fields should be 1) search_query (make it a list of possible queries which can find the best items for the user in Divar website) 2) price_range with min and max number In Tomans 3) possible_filters (keys must be in english)' },    
         )  
         response = get_completion_from_messages(messages, temperature=0)
-        # save = response
         query = json.loads(response)["search_query"][0]
         h = f"<iframe src='https://www.google.com/search?igu=1&q=site:divar.ir+{query}' width='100%' height='400' title='description'></iframe>"
     except Exception as e:
         h, response = "WRONG API KEY", str(e)
-    return "", chat_history, h, response
+    return "", chat_history, h, response, context
 
 def clear_fn():
-    global context
-    context = base_context.copy()
+    return None, base_context.copy()
 
 with gr.Blocks() as demo:
     with gr.Row():
@@ -84,7 +82,9 @@ with gr.Blocks() as demo:
             chatbot = gr.Chatbot()
             msg = gr.Textbox(label="Chat")
             clear = gr.Button("Clear")
-            msg.submit(respond, [msg, chatbot, api_key_box], [msg, chatbot, html, htmlj])
-            clear.click(clear_fn, None, chatbot, queue=False)
+            context_box = gr.Textbox(base_context.copy(), label="Context", visible=False)
+            msg.submit(respond, [msg, chatbot, api_key_box, context_box], [msg, chatbot, html, htmlj, context_box])
+            clear.click(clear_fn, None, [chatbot, context_box], queue=False)
+            
 
 demo.launch()