Update app.py

app.py

@@ -45,13 +45,19 @@ async def ask(ctx, *, question: str):
     prompt = f"Please provide a detailed response to the following question: {question}"
     await ctx.send(f"Question: {question}")
     response = query_huggingface(prompt)
-    if 'generated_text' in response:
-        await ctx.send(f"Response: {response['generated_text']}")
+    
+    # Extract and clean the response
+    if isinstance(response, dict) and 'generated_text' in response:
+        response_text = response['generated_text']
     elif isinstance(response, list) and len(response) > 0 and 'generated_text' in response[0]:
-        await ctx.send(f"Response: {response[0]['generated_text']}")
+        response_text = response[0]['generated_text']
     else:
-        await ctx.send("Sorry, I couldn't generate a response.")
+        response_text = "Sorry, I couldn't generate a response."
+    
+    # Remove the prompt from the response if present
+    clean_response = response_text.replace(prompt, '').strip()
+    
+    await ctx.send(f"Response: {clean_response}")
 
 # Run the bot
 bot.run(DISCORD_TOKEN)
-