Update app.py

app.py

@@ -45,6 +45,18 @@ DIVAR_CONTEXT = """
 برای کسب‌وکار : خرید و فروش تجهیزات و ماشین‌آلات، عمده‌فروشی و ...
 استخدام و کاریابی : اداری و مدیریتی، سرایداری، عمران و ساختمانی، رایانه و فناوری اطلاعات، مالی و حسابداری و ...
 شما باید کوتاه و بسیار دوستانه پاسخ دهید.
+
+Each answer from the bot must have two parts separated by "---". The first part is the response to the user. The second part is a json.
+In the second part 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 in Farsi. If there are none put an empty string in the list.) 2) price_range with min and max number In Tomans 3) possible_filters (keys must be in english)
+The results of this query will be shown to the user next to the chat by the website.
+As you do not have the recent information such as prices, you can refer the user for such information to the results of the query that are shown on the left side of their chat.
+
+Example:
+سلام من ربات دستیار دیوار هستم.
+---
+{"search_query": ["131 پراید نو"], "price_range": [0, 100000], "possible_filters": {"brand": "پراید", "model": "131"}}
+
+Make sure that every response has these two parts separated by "---" and the json is a valid json.
 """
 
 base_context = [ {'role':'system', 'content': DIVAR_CONTEXT} ]  # accumulate messages
@@ -56,23 +68,27 @@ def respond(message, chat_history, api_key, context_box):
         openai.api_key = api_key
         save = api_key
         bot_message, context = collect_messages(message, context)
-        chat_history.append((message, bot_message))
+        user_response = bot_message.split("---")[0]
+        query_response = bot_message.split("---")[1]
+        chat_history.append((message, user_response))
         
-        messages = context.copy()
-        messages.append(
-            {'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)
-        query = json.loads(response)["search_query"][0]
+        # messages = context.copy()
+        # messages.append(
+        #     {'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)
+        query = ast.literal_eval(query_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, context
+        h, query_response = bot_message, str(e)
+    return "", chat_history, h, query_response, context
 
 def clear_fn():
     return None, base_context.copy()
 
-with gr.Blocks() as demo:
+with gr.Blocks(title="DivarGPT") as demo:
+    with gr.Row():
+        gr.HTML('<h2 style="text-align: center; margin-bottom: 1rem">DivarGPT v0.2</h1>')
     with gr.Row():
         with gr.Column(scale=1):
             html = gr.HTML("Results Will be shown here!")
@@ -86,5 +102,4 @@ with gr.Blocks() as demo:
             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()