Added code to wait for Hope to wake up if it is sleeping

app.py

@@ -27,7 +27,15 @@ hope_client = Client("HopeLLM/HopeV1.0", hf_token=os.getenv("HF_READ_TKN"))
 Lock = threading.Lock()
 TIMESTAMP_LAST_CALL_TO_HOPE = None
 
-def call_hope_api(input_txt):
+def is_space_sleeping():
+    headers = {"Authorization": f"Bearer {os.getenv("HF_READ_TKN")}"}
+    response = requests.get("https://huggingface.co/api/spaces/HopeLLM/HopeV1.0", headers=headers)
+    if response.status_code == 200:
+        space_info = response.json()
+        return space_info.get("runtime", {}).get("stage", "") == "sleeping"
+    return "unknown"  # If API call fails, assume the space is sleeping
+
+def call_hope_api(input_txt, sender_id):
     global hope_client
     global TIMESTAMP_LAST_CALL_TO_HOPE
 
@@ -35,6 +43,14 @@ def call_hope_api(input_txt):
     with Lock:
         if hope_client == None:
             hope_client = Client("HopeLLM/HopeV1.0", hf_token=os.getenv("HF_READ_TKN"))
+        if(is_space_sleeping()==True):
+            logger.info("Hope is sleeping, trying to wake it up")
+            send_whatsapp_message(sender_id, None, "Hope is sleeping, waking it up ;)\nKindly wait for some time", text=True)
+            while(is_space_sleeping()):
+                #Sleep for 30sec
+                time.sleep(30)
+        elif (is_space_sleeping() == "unknown"):
+            logger.error("Something wrong with Hope Check")
         return hope_client.predict(inp = input_txt, api_name = "/answer_question") 
 
 def reset_hope_client():
@@ -142,13 +158,13 @@ def backgnd_process_message(message):
                 response_text = "Awesome, got your story. Now, let me ask you some questions."
                 ## Generate question from llama 
                 llama_input  = llama_input_template.format(story_to_use = message_text, fixed_prompt=FIXED_PROMPT)
-                llama_output = call_hope_api(llama_input)
+                llama_output = call_hope_api(llama_input,sender_id)
                 logger.info(f"LLama Ouput: {llama_output}")
                 response_text = response_text + "My question for you is \n \n" + llama_output
                 ONGOING_CONVO[sender_id]['Llama_conv'] = llama_input + llama_output + "</s> <s> [INST] "
             else:
                 llama_input = ONGOING_CONVO[sender_id]['Llama_conv'] + message_text + "[/INST]"
-                llama_output = call_hope_api(llama_input)
+                llama_output = call_hope_api(llama_input,sender_id)
                 logger.info(f"LLama Ouput: {llama_output}")
                 response_text = llama_output
                 ONGOING_CONVO[sender_id]['Llama_conv'] = llama_input + llama_output + "</s> <s> [INST] "
@@ -162,13 +178,13 @@ def backgnd_process_message(message):
                 response_text = "Awesome, got your story. Now, let me ask you some questions."
                 ## Generate question from llama 
                 llama_input  = llama_input_template.format(story_to_use = transcription, fixed_prompt=FIXED_PROMPT)
-                llama_output = call_hope_api(llama_input)
+                llama_output = call_hope_api(llama_input,sender_id)
                 logger.info(f"LLama Ouput: {llama_output}")
                 response_text = response_text + "My question for you is \n \n" + llama_output
                 ONGOING_CONVO[sender_id]['Llama_conv'] = llama_input + llama_output + "</s> <s> [INST] "
             else:
                 llama_input = ONGOING_CONVO[sender_id]['Llama_conv'] + transcription + "[/INST]"
-                llama_output = call_hope_api(llama_input)
+                llama_output = call_hope_api(llama_input,sender_id)
                 logger.info(f"LLama Ouput: {llama_output}")
                 response_text = llama_output
                 ONGOING_CONVO[sender_id]['Llama_conv'] = llama_input + llama_output + "</s> <s> [INST] "
@@ -176,13 +192,13 @@ def backgnd_process_message(message):
             if message["interactive"]["button_reply"]["id"] == "random_story":
                 ONGOING_CONVO[sender_id]['Story'] = "random_story"
                 #send_whatsapp_message(sender_id, message_id, "Perfect, let me find a nice story for today.")
-                story_for_chat = call_hope_api(llama_input_gen_random_story)
+                story_for_chat = call_hope_api(llama_input_gen_random_story,sender_id)
                 ONGOING_CONVO[sender_id]['Story'] = story_for_chat
                 logger.info(f"Llama gave story {story_for_chat}")
 
                 ## Generate question from llama 
                 llama_input  = llama_input_template.format(story_to_use = story_for_chat, fixed_prompt=FIXED_PROMPT)
-                llama_output = call_hope_api(llama_input)
+                llama_output = call_hope_api(llama_input,sender_id)
                 logger.info(f"LLama Ouput: {llama_output}")
                 response_text = story_for_chat + "\n \n" + llama_output
                 ONGOING_CONVO[sender_id]['Llama_conv'] = llama_input + llama_output + "</s> <s> [INST] "