Update app.py

app.py

@@ -44,20 +44,34 @@ async def generate_command(
         genai.configure(api_key=GEMINI_API_KEY)
         model = genai.GenerativeModel("gemini-2.0-flash-exp")
         response = model.generate_content(prompt, stream=True)
+        current_message = None  # Initialize outside the loop to keep track of current message
         current_message_content = ""
+
         for part in response:
             text_chunk = part.text
 
             if len(current_message_content) + len(text_chunk) <= 2000:
                 current_message_content += text_chunk
+                if current_message:
+                    await current_message.edit(content=current_message_content)
+                else:
+                    current_message = await interaction.followup.send(content=current_message_content)
             else:
                 # Send the current message
-                await interaction.followup.send(content=current_message_content)
+                if current_message:
+                  await current_message.edit(content=current_message_content)
+                else:
+                  await interaction.followup.send(content=current_message_content)
                 current_message_content = text_chunk  # Start new message with the current chunk
+                current_message = await interaction.followup.send(content=current_message_content)
+
 
         # Send any remaining content
         if current_message_content:
-            await interaction.followup.send(content=current_message_content)
+           if current_message:
+             await current_message.edit(content=current_message_content)
+           else:
+             await interaction.followup.send(content=current_message_content)
     except Exception as e:
         print(e)