[falcon.py] debugging messages

falcon.py

@@ -27,9 +27,8 @@ async def try_falcon(interaction, prompt):
 
         if interaction.user.id != BOT_USER_ID:
             if interaction.channel.id == FALCON_CHANNEL_ID: 
-
                 if os.environ.get('TEST_ENV') == 'True':
-                    print("TEST_ENV is True.")
+                    print("Safetychecks passed for try_falcon")
                 await interaction.response.send_message(f"Working on it!")
                 channel = interaction.channel
                 message = await channel.send(f"Creating thread...")
@@ -38,10 +37,14 @@ async def try_falcon(interaction, prompt):
                 f"model and system prompt can be found here: https://huggingface.co/spaces/HuggingFaceH4/falcon-chat]")
                 
                 chathistory = falcon_client.predict(fn_index=5) 
+                if os.environ.get('TEST_ENV') == 'True':
+                    print("Running falcon_client.submit")
                 job = falcon_client.submit(prompt, chathistory, instructions, 0.8, 0.9, fn_index=1)  # This is not blocking, similar to run_in_executor (but better)
                 while job.done() is False: 
                     pass
                 else:
+                    if os.environ.get('TEST_ENV') == 'True':
+                        print("Try_falcon job done")
                     file_paths = job.outputs()
                     full_generation = file_paths[-1]
                 with open(full_generation, 'r') as file:
@@ -62,17 +65,22 @@ async def continue_falcon(message):
         if not message.author.bot:
             global falcon_userid_threadid_dictionary # tracks userid-thread existence
             if message.channel.id in falcon_userid_threadid_dictionary: # is this a valid thread?
-                 if falcon_userid_threadid_dictionary[message.channel.id] == message.author.id: # more than that - is this specifically the right user for this thread?
+                if falcon_userid_threadid_dictionary[message.channel.id] == message.author.id: # more than that - is this specifically the right user for this thread?
+                    if os.environ.get('TEST_ENV') == 'True':
+                        print("Safetychecks passed for continue_falcon")                     
                     global instructions
                     global threadid_conversation
                     await message.add_reaction('🔁') 
                     chathistory = threadid_conversation[message.channel.id]
                     prompt = message.content
-
+                    if os.environ.get('TEST_ENV') == 'True':
+                        print("Running falcon_client.submit")
                     job = falcon_client.submit(prompt, chathistory, instructions, 0.8, 0.9, fn_index=1)
                     while job.done() is False: 
                         pass
                     else:
+                        if os.environ.get('TEST_ENV') == 'True':
+                            print("Continue_falcon job done")
                         file_paths = job.outputs()
                         full_generation = file_paths[-1]
                     with open(full_generation, 'r') as file: