Update falcon.py

falcon.py

@@ -6,6 +6,7 @@ falcon_client = info.falcon_client
 json = info.json
 threadid_conversation = info.threadid_conversation
 falcon_userid_threadid_dictionary = info.falcon_userid_threadid_dictionary
+instructions = info.instructions
 
 async def falcon_command(interaction: discord.Interaction, prompt: str):
     try:  
@@ -21,7 +22,7 @@ async def falcon_command(interaction: discord.Interaction, prompt: str):
                 chathistory = falcon_client.predict( 
                         fn_index=5
                 ) # []    
-                job = falcon_client.submit(prompt, chathistory, info.instructions, 0.8, 0.9, fn_index=1)  # This is not blocking, similar to run_in_executor (but better)
+                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() == False: 
                     status = job.status() 
                     #print(status)
@@ -38,4 +39,32 @@ async def falcon_command(interaction: discord.Interaction, prompt: str):
                 await thread.send(f"{output_text}")  
 
     except Exception as e:
-        print(f"Error: {e}")
+        print(f"Error: {e}")
+
+async def continue_falcon(message):
+    """continues a given conversation based on chathistory"""
+    try:
+        await message.add_reaction('<a:loading:1114111677990981692>') 
+        chathistory = threadid_conversation[message.channel.id]
+        prompt = message.content
+        # generation
+        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() == False: 
+            status = job.status() 
+            #print(status)
+        else:
+            file_paths = job.outputs()
+            full_generation = file_paths[-1] # tmp12345678.json
+        with open(full_generation, 'r') as file:
+            data = json.load(file)
+            output_text = data[-1][-1] # we output this as the bot
+
+        threadid_conversation[message.channel.id] = full_generation # overwrite the old file
+        falcon_userid_threadid_dictionary[message.channel.id] = message.author.id
+        print(output_text)
+        await message.reply(f"{output_text}")  
+        await message.remove_reaction('<a:loading:1114111677990981692>', client.user) 
+
+    except Exception as e:
+        print(f"Error: {e}")
+        await message.reply(f"Error: {e} <@811235357663297546> (continue_falcon error)")