ShareGPT for index and code

cogs/code_interpreter_service_cog.py

@@ -133,13 +133,15 @@ async def on_message(self, message):
                 return
 
         prompt = message.content.strip()
-
+        self.converser_cog.full_conversation_history[message.channel.id].append(prompt)
         # If the message channel is in self.chat_agents, then we delegate the message to the agent.
         if message.channel.id in self.chat_agents:
             if prompt.lower() in ["stop", "end", "quit", "exit"]:
                 await message.reply(
-                    "Ending chat session. You can access the sandbox of this session at https://"
-                    + self.sessions[message.channel.id].get_hostname()
+                    content = "Ending chat session. You can access the sandbox of this session at https://"
+                    + self.sessions[message.channel.id].get_hostname(),
+                    view = ShareView(self.converser_cog, message.channel.id) 
+                    if isinstance(message.channel, discord.Thread) else None
                 )
                 self.sessions[message.channel.id].close()
                 self.chat_agents.pop(message.channel.id)
@@ -233,7 +235,6 @@ async def on_message(self, message):
                     print(stdout_output)
                 except:
                     pass
-
             except Exception as e:
                 response = f"Error: {e}"
                 traceback.print_exc()
@@ -242,7 +243,7 @@ async def on_message(self, message):
                 )
                 safe_remove_list(self.thread_awaiting_responses, message.channel.id)
                 return
-
+            self.converser_cog.full_conversation_history[message.channel.id].append(response)
             # Parse the artifact names. After Artifacts: there should be a list in form [] where the artifact names are inside, comma separated inside stdout_output
             artifact_names = re.findall(r"Artifacts: \[(.*?)\]", stdout_output)
             # The artifacts list may be formatted like ["'/home/user/artifacts/test2.txt', '/home/user/artifacts/test.txt'"], where its technically 1 element in the list, so we need to split it by comma and then remove the quotes and spaces
@@ -568,3 +569,61 @@ async def callback(self, interaction: discord.Interaction):
                 await self.ctx.channel.send(
                     "Failed to download artifact: " + artifact, delete_after=120
                 )
+
+class ShareView(discord.ui.View):
+    def __init__(
+        self,
+        converser_cog,
+        conversation_id,
+    ):
+        super().__init__(timeout=3600)  # 1 hour interval to share the conversation.
+        self.converser_cog = converser_cog
+        self.conversation_id = conversation_id
+        self.add_item(ShareButton(converser_cog, conversation_id))
+
+    async def on_timeout(self):
+        # Remove the button from the view/message
+        self.clear_items()
+
+
+class ShareButton(discord.ui.Button["ShareView"]):
+    def __init__(self, converser_cog, conversation_id):
+        super().__init__(
+            style=discord.ButtonStyle.green,
+            label="Share Conversation",
+            custom_id="share_conversation",
+        )
+        self.converser_cog = converser_cog
+        self.conversation_id = conversation_id
+
+    async def callback(self, interaction: discord.Interaction):
+        # Get the user
+        try:
+            id = await self.converser_cog.sharegpt_service.format_and_share(
+                self.converser_cog.full_conversation_history[self.conversation_id],
+                self.converser_cog.bot.user.default_avatar.url
+                if not self.converser_cog.bot.user.avatar
+                else self.converser_cog.bot.user.avatar.url,
+            )
+            url = f"https://shareg.pt/{id}"
+            await interaction.response.send_message(
+                embed=EmbedStatics.get_conversation_shared_embed(url)
+            )
+        except ValueError as e:
+            traceback.print_exc()
+            await interaction.response.send_message(
+                embed=EmbedStatics.get_conversation_share_failed_embed(
+                    "The ShareGPT API returned an error: " + str(e)
+                ),
+                ephemeral=True,
+                delete_after=15,
+            )
+            return
+        except Exception as e:
+            traceback.print_exc()
+            await interaction.response.send_message(
+                embed=EmbedStatics.get_conversation_share_failed_embed(str(e)),
+                ephemeral=True,
+                delete_after=15,
+            )
+            return

cogs/index_service_cog.py

@@ -13,7 +13,9 @@
 from services.environment_service import EnvService
 from services.moderations_service import Moderation
 from services.text_service import TextService
+from services.sharegpt_service import ShareGPTService
 from models.index_model import Index_handler
+from collections import defaultdict
 from utils.safe_ctx_respond import safe_remove_list
 
 USER_INPUT_API_KEYS = EnvService.get_user_input_api_keys()
@@ -36,8 +38,10 @@ def __init__(
         super().__init__()
         self.bot = bot
         self.index_handler = Index_handler(bot, usage_service)
+        self.full_conversation_history = defaultdict(list)
         self.thread_awaiting_responses = []
         self.deletion_queue = deletion_queue
+        self.sharegpt_service = ShareGPTService()
 
     async def process_indexing(self, message, index_type, content=None, link=None):
         """
@@ -82,6 +86,8 @@ async def process_indexing(self, message, index_type, content=None, link=None):
             await message.reply(embed=failure_embed)
             safe_remove_list(self.thread_awaiting_responses, message.channel.id)
             return False
+        # summary is type casted to string because it might be a Response object
+        self.full_conversation_history[message.channel.id].append(str(summary))
 
         success_embed = discord.Embed(
             title=f"{index_type.capitalize()} Interpreted",
@@ -129,6 +135,8 @@ async def on_message(self, message):
 
         prompt = message.content.strip()
 
+        self.full_conversation_history[message.channel.id].append(prompt)
+
         if await self.index_handler.get_is_in_index_chat(message):
             self.thread_awaiting_responses.append(message.channel.id)
 
@@ -139,6 +147,19 @@ async def on_message(self, message):
 
             # Handle file uploads
             file = message.attachments[0] if len(message.attachments) > 0 else None
+
+            if prompt.lower() in ["stop", "end", "quit", "exit"]:
+                await message.reply(
+                view = ShareView(self, message.channel.id) if isinstance(message.channel, discord.Thread) else None
+                )
+                await message.reply("Ending chat session.")
+                self.index_handler.index_chat_chains.pop(message.channel.id)
+
+                # close the thread
+                thread = await self.bot.fetch_channel(message.channel.id)
+                await thread.edit(name="Closed-GPT")
+                await thread.edit(archived=True)
+                return "Ended chat session."
 
             # File operations, allow for user file upload
             if file:
@@ -186,6 +207,8 @@ async def on_message(self, message):
                 )
                 return
 
+            self.full_conversation_history[message.channel.id].append(chat_result)
+
             if chat_result:
                 if len(chat_result) > 2000:
                     embed_pages = EmbedStatics.paginate_chat_embed(chat_result)
@@ -492,3 +515,61 @@ async def compose_command(self, ctx, name):
                 return
 
         await self.index_handler.compose(ctx, name, user_api_key)
+
+class ShareView(discord.ui.View):
+    def __init__(
+        self,
+        converser_cog,
+        conversation_id,
+    ):
+        super().__init__(timeout=3600)  # 1 hour interval to share the conversation.
+        self.converser_cog = converser_cog
+        self.conversation_id = conversation_id
+        self.add_item(ShareButton(converser_cog, conversation_id))
+
+    async def on_timeout(self):
+        # Remove the button from the view/message
+        self.clear_items()
+
+
+class ShareButton(discord.ui.Button["ShareView"]):
+    def __init__(self, indexer_cog, conversation_id):
+        super().__init__(
+            style=discord.ButtonStyle.green,
+            label="Share Conversation",
+            custom_id="share_conversation",
+        )
+        self.indexer_cog = indexer_cog
+        self.conversation_id = conversation_id
+
+    async def callback(self, interaction: discord.Interaction):
+        # Get the user
+        try:
+            id = await self.indexer_cog.sharegpt_service.format_and_share(
+                self.indexer_cog.full_conversation_history[self.conversation_id],
+                self.indexer_cog.bot.user.default_avatar.url
+                if not self.indexer_cog.bot.user.avatar
+                else self.indexer_cog.bot.user.avatar.url,
+            )
+            url = f"https://shareg.pt/{id}"
+            await interaction.response.send_message(
+                embed=EmbedStatics.get_conversation_shared_embed(url)
+            )
+        except ValueError as e:
+            traceback.print_exc()
+            await interaction.response.send_message(
+                embed=EmbedStatics.get_conversation_share_failed_embed(
+                    "The ShareGPT API returned an error: " + str(e)
+                ),
+                ephemeral=True,
+                delete_after=15,
+            )
+            return
+        except Exception as e:
+            traceback.print_exc()
+            await interaction.response.send_message(
+                embed=EmbedStatics.get_conversation_share_failed_embed(str(e)),
+                ephemeral=True,
+                delete_after=15,
+            )
+            return

models/index_model.py

@@ -375,6 +375,7 @@ async def execute_index_chat_message(self, ctx, message):
         if ctx.channel.id not in self.index_chat_chains:
             return None
 
+        '''Moved this logic to index_service_cog.py
         if message.lower() in ["stop", "end", "quit", "exit"]:
             await ctx.reply("Ending chat session.")
             self.index_chat_chains.pop(ctx.channel.id)
@@ -384,6 +385,7 @@ async def execute_index_chat_message(self, ctx, message):
             await thread.edit(name="Closed-GPT")
             await thread.edit(archived=True)
             return "Ended chat session."
+        '''
 
         self.usage_service.update_usage_memory(ctx.guild.name, "index_chat_message", 1)