Add function to search for 3 or more mediators

app.py

@@ -36,6 +36,10 @@ class MediatorRetriever(BaseRetriever):
                 "parameters": {
                     "type": "object",
                     "properties": {
+                        "mediator state": {
+                            "type": "string",
+                            "description": "Extract mediator's state that user wants to search. If both mediator city and mediator state are possible, please extract as mediator state."
+                        },
                         "mediator country": {
                             "type": "string",
                             "description": "Extract mediator's country that user wants to search."
@@ -44,10 +48,6 @@ class MediatorRetriever(BaseRetriever):
                             "type": "string",
                             "description": "Extract mediator's city that user wants to search."
                         },
-                        "mediator state": {
-                            "type": "string",
-                            "description": "Extract mediator's state that user wants to search. If both mediator city and mediator state are possible, please extract as mediator state."
-                        },
                         "mediator areas of practice": {
                             "type": "array",
                             "description": description,
@@ -61,7 +61,7 @@ class MediatorRetriever(BaseRetriever):
         response = openai_client.chat.completions.create(
             model="gpt-4-1106-preview",
             messages=[
-                {"role": "system", "content": "You are a professional mediation field searcher. Your role is to extract information about mediators from user's message. but you have to provide several mediators not one. If there are no mediators in the specific city, then it should return mediators from the state."},
+                {"role": "system", "content": "You are a professional arbitration field searcher. Your role is to extract information about the moderator from the user's message. If you have less than 3 mediators in a particular city, you should search for mediators in your state. If you have less than 3 mediators in a particular state that you are searching for, you should search for mediators in the states closest to the state that you are searching for.  For example, if you have less than 3 mediators in California, you should search for mediators in Oregon, Arizona, or Nevada closest to California. Important: You must provide at least three moderators."},
                 {"role": "user", "content": message}
             ],
             functions=tools,
@@ -74,7 +74,6 @@ class MediatorRetriever(BaseRetriever):
                 yield {"search_status" : False, "data" : delta.content}
             elif delta and delta.function_call:
                 new_message += delta.function_call.arguments
-        
         if new_message:
             yield {"search_status" : True, "data" : new_message}
 
@@ -83,6 +82,7 @@ class MediatorRetriever(BaseRetriever):
         matching_documents = []
         message = ""
         metaData = self.getMetadata(query)
+        
         for data in metaData:
             if data['data'] is None:
                 break
@@ -112,7 +112,7 @@ class MediatorRetriever(BaseRetriever):
                     response = openai_client.chat.completions.create(
                         model="gpt-4-1106-preview",
                         messages=[
-                            {"role": "system", "content": "Generate a message asking the user about the conflict to better match them with a mediator."},
+                            {"role": "system", "content": "Generate a message asking the user about the conflict to better match them with mediators."},
                             {"role": "user", "content": query}
                         ],
                         stream=True
@@ -150,7 +150,7 @@ class MediatorRetriever(BaseRetriever):
                                 "properties": {
                                     "mediator": {
                                         "type": "number",
-                                        "description": "The number of mediators that the user wants to search. If the user asks for a list of mediators, it means the user wants to search at least 3 mediators. If the user's message doesn't have information about the number of mediators, you have to respond with at least 3.",
+                                        "description": "The number of mediators the user wants to search for. If a user requests a list of mediators, this means that the user wants to search for at least three mediators. If your message does not include information about the number of mediators, you must provide at least three mediators. If you only have one arbitrator, you must search three mediators in the state and out of state. In example, If there is no three mediators in California search the mediators from Oregon, Arizona, or Nevada. So you must provide at least three mediators. IMPORTANT: you must provide at least three mediators.",
                                         "default": 3
                                     }
                                 },
@@ -169,7 +169,7 @@ class MediatorRetriever(BaseRetriever):
                         stream=True
                     )
                     try:
-                        number_str = response.choices[0].message.tool_calls[0].function.arguments
+                        number_str = response.choices[0].message.function_call.arguments
                         mediator_num = max(int(json.loads(number_str)['mediator']), 3)
                     except:
                         mediator_num = 3
@@ -193,7 +193,7 @@ class MediatorRetriever(BaseRetriever):
                     embeddings = OpenAIEmbeddings(api_key=openai_api_key)
                     
                     index = pc.Index(pinecone_index)
-
+                        
                     results = index.query(
                         vector=embeddings.embed_query(query),
                         top_k=5,
@@ -201,24 +201,17 @@ class MediatorRetriever(BaseRetriever):
                         include_metadata=True
                     )
                     
-                    if (not results['matches']) and 'mediator state' in metadata:
-                        del metadata['mediator city']
-                        results = index.query(
-                            vector=embeddings.embed_query(query),
-                            top_k=5,
-                            filter=metadata,
-                            include_metadata=True
-                        )
-
                     new_data = []
                     for result in results['matches']:
                         data = {}
                         for metadata_key in metadata_list:      
                             data[metadata_key] = result['metadata'].get(metadata_key, "N/A")
+                        # practice_data = result['metadata'].get('mediator areas of practice', "N/A")
+                        if not practice_data:
+                            continue
                         
-                        if practice_data in result['metadata'].get('mediator areas of practice', []):
+                        if practice_data:
                             new_data.append(data)
-
                     random.shuffle(new_data)
 
                     if new_data:
@@ -298,7 +291,7 @@ chatbot = gr.Chatbot(avatar_images=["user.png", "bot.jpg"], height="100%",contai
 
 js_code = """
 <style>
-     html body {
+    html body {
         background: transparent !important;
     }
     
@@ -352,8 +345,8 @@ js_code = """
         border-radius: 9px;
         height: 100%;
         opacity: 0;
-        transform: scaleY(0); /* Start with scaleY 0 */
-        transform-origin: bottom; /* Transform origin at bottom */
+        transform: scaleY(0); 
+        transform-origin: bottom; 
         transition: transform 0.5s ease-out, opacity 0.5s ease-out;
         overflow: hidden;
         position:fixed;
@@ -376,15 +369,16 @@ js_code = """
         border-bottom-left-radius: 0px;
         border-bottom-right-radius: 0px;
     }
-    .gradio-container {
+    .gradio-container{
         padding:0 !important;
+        background-color:transparent;
     }
-
+    
     gradio-app {
         background: none !important;
     }
     
- </style>
+</style>
 <script>
     function toggleChatbox() {
         window.parent.postMessage(
@@ -406,33 +400,25 @@ js_code = """
 
     
     function autoScroll() {
-        const chatbox = document.querySelector('.bubble-wrap');
+        const chatbox = document.querySelector('#component-4');
         if (chatbox) {
             chatbox.scrollTop = chatbox.scrollHeight;
         }
     }
 
     setTimeout(() => {
-       document.getElementById('message-btn').addEventListener('click', toggleChatbox);
+        document.getElementById('message-btn').addEventListener('click', toggleChatbox);
         const userInputBox = document.querySelector('textarea.scroll-hide');
         userInputBox?.addEventListener('keypress', autoScroll, false);
-    }, 2000);   
+    }, 1000);   
 </script>
 """
 
-def toggle_chatbox():
-    global chatbox_visible
-    chatbox_visible = not chatbox_visible
-    return gr.update(visible=chatbox_visible)
-
-chatbox_visible = True
-
 with gr.Blocks(head=js_code) as demo:
-    with gr.Column(visible=chatbox_visible) as chatbox_container:
+    with gr.Column() as chatbox_container:
         chatbox = gr.ChatInterface(fn=search, title="Mediate.com Chatbot Application", multimodal=False, retry_btn=None, clear_btn=None, undo_btn=None, chatbot=chatbot)
     
-    # Add an event listener to the button
-    gr.HTML(f'<button src="/file=message.png" id="message-btn"/><img style="display: inline;" src="/file=message.png" id="custom-btn"/><button>')
+    gr.HTML(f'<button id="message-btn"/><img style="display: inline;" src="/file=message.png" id="custom-btn"/><button>')
 
 if __name__ == "__main__":
     demo.launch(debug=True, allowed_paths=["./"])