Update app.py

app.py

@@ -67,6 +67,26 @@ def enable_components(recognized_text):
     loading_animation_update = gr.update(visible=False)
     return recognized_text, process_button_update, loading_animation_update
 
+# Define a function to disable the button and display a loading indicator
+def disable_chatbot_components():
+    textbox = gr.update(interactive=False)
+    submit_btn = gr.update(interactive=False)
+    btn1 = gr.update(interactive=False)
+    btn2 = gr.update(interactive=False)
+    btn3 = gr.update(interactive=False)
+    btn4 = gr.update(interactive=False)
+    return textbox, submit_btn, btn1, btn2, btn3, btn4
+
+# Define a function to enable the button and hide the loading indicator
+def enable_chatbot_components():
+    textbox = gr.update(interactive=True)
+    submit_btn = gr.update(interactive=True)
+    btn1 = gr.update(interactive=True)
+    btn2 = gr.update(interactive=True)
+    btn3 = gr.update(interactive=True)
+    btn4 = gr.update(interactive=True)
+    return textbox, submit_btn, btn1, btn2, btn3, btn4
+
 llama_responded = 0
 responded_answer = ""
 
@@ -76,30 +96,62 @@ def respond(
 ):
     global llama_responded
     global responded_answer
-    system_message = "You are a helpful chatbot that answers questions. Give any answer within 50 words."
-    messages = [{"role": "system", "content": system_message}]
-
-    for val in history:
-        print(val[0])
-        if val[0] != None:
-            if val[0]:
-                messages.append({"role": "user", "content": val[0]})
-            if val[1]:
-                messages.append({"role": "assistant", "content": val[1]})
-    messages.append({"role": "user", "content": message})
-
-    response = ""
-
-    for message in client.chat_completion(
-        messages,
+    # Main Decision Module
+    decision_response = ""
+    judge_main_message = f"Here is a query: '{message}', Determine if this query is asking about one of the topics included in the list below. If it is, please directly provide only one name of the topic; otherwise, you reply 'no'. The list of topics is: [movie, music]"
+    m_message = [{"role": "user", "content": judge_main_message}]
+    for m in client.chat_completion(
+        m_message,
         stream=True,
     ):
-        token = message.choices[0].delta.content
-        response += token
+        token = m.choices[0].delta.content
+        decision_response += token
+    print(decision_response)
+
+    if "movie" in decision_response:
+        movie_client = Client("ironserengety/movies-recommender")
+        result = movie_client.predict(
+                message=message,
+                system_message="You are a movie recommender named 'Exodia'. You are extremely reliable. You always mention your name in the beginning of conversation. You will provide me with answers from the given info. Give not more than 3 choices and make sure that answers are complete sentences.",
+                max_tokens=512,
+                temperature=0.7,
+                top_p=0.95,
+                api_name="/chat"
+        )
+        print(result)
+        llama_responded = 1
+        responded_answer = result
+        return result
+    
+    #elif "music" in decision_response:
+
+    else:
+        #others
+        system_message = "You are a helpful chatbot that answers questions. Give any answer within 50 words."
+        messages = [{"role": "system", "content": system_message}]
+
+        for val in history:
+            print(val[0])
+            if val[0] != None:
+                if val[0]:
+                    messages.append({"role": "user", "content": val[0]})
+                if val[1]:
+                    messages.append({"role": "assistant", "content": val[1]})
+        messages.append({"role": "user", "content": message})
+
+        response = ""
+        print(messages)
 
-    llama_responded = 1
-    responded_answer = response
-    return response
+        for message in client.chat_completion(
+            messages,
+            stream=True,
+        ):
+            token = message.choices[0].delta.content
+            response += token
+
+        llama_responded = 1
+        responded_answer = response
+        return response
 
 def update_response_display():
     while not llama_responded:
@@ -181,21 +233,36 @@ def create_interface():
         )
 
         user_start.then(
+            fn=disable_chatbot_components,
+            inputs=[],
+            outputs=[chatbot.submit_btn, chatbot.textbox, process_button, chatbot.retry_btn, chatbot.undo_btn, chatbot.clear_btn]
+        ).then(
             fn=tts_part,
             inputs=[], 
             outputs=text_speaker
+        ).then(
+            fn=enable_chatbot_components,
+            inputs=[],
+            outputs=[chatbot.submit_btn, chatbot.textbox, process_button, chatbot.retry_btn, chatbot.undo_btn, chatbot.clear_btn]
         )
 
         user_click.then(
+            fn=disable_chatbot_components,
+            inputs=[],
+            outputs=[chatbot.submit_btn, chatbot.textbox, process_button, chatbot.retry_btn, chatbot.undo_btn, chatbot.clear_btn]
+        ).then(
             fn=tts_part,
             inputs=[], 
             outputs=text_speaker
+        ).then(
+            fn=enable_chatbot_components,
+            inputs=[],
+            outputs=[chatbot.submit_btn, chatbot.textbox, process_button, chatbot.retry_btn, chatbot.undo_btn, chatbot.clear_btn]
         )
                 
     return demo
 
 
-
 if __name__ == "__main__":
     demo = create_interface()
     demo.launch()