Update app.py

app.py

@@ -34,7 +34,8 @@ def query_huggingface(prompt, temperature=0.5, max_tokens=100, top_k=50, top_p=0
     try:
         response = requests.post(API_URL, headers=headers, json=payload)
         response.raise_for_status()
-        return response.json()
+        result = response.json()
+        return result[0]['generated_text'] if isinstance(result, list) and len(result) > 0 else result.get('generated_text', 'No response generated.')
     except requests.exceptions.RequestException as e:
         logger.error(f"Error querying the API: {e}")
         return {"error": str(e)}
@@ -67,23 +68,14 @@ async def ask(ctx, question: str, temperature: float = 0.8, max_tokens: int = 10
         return
 
     # Create a structured prompt
-    prompt = f"Do not under any circumstance, finish or add to the input. Excluded input correction or finishing at all times. Answer the following question as accurately and concisely as possible: {question}"
+    prompt = f"Answer the following question accurately and concisely: {question}"
     response = query_huggingface(prompt, temperature, max_tokens, top_k, top_p)
     
     # 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]:
-        response_text = response[0]['generated_text']
+    if 'error' in response:
+        clean_response = f"Error: {response['error']}"
     else:
-        response_text = "Sorry, I couldn't generate a response."
-
-    # Remove the prompt from the response if present
-    clean_response = response_text.replace(prompt, '').strip()
-    
-    # Avoid prompt completion issues by removing any leading incomplete sentence
-    if clean_response.startswith(question):
-        clean_response = clean_response[len(question):].strip()
+        clean_response = response.replace(prompt, '').strip()
     
     await ctx.send(clean_response)
 
@@ -91,6 +83,9 @@ async def ask(ctx, question: str, temperature: float = 0.8, max_tokens: int = 10
 async def ask_error(ctx, error):
     if isinstance(error, commands.CommandOnCooldown):
         await ctx.send(f"This command is on cooldown. Please try again after {int(error.retry_after)} seconds.")
+    else:
+        logger.error(f"Error in ask command: {error}")
+        await ctx.send("An error occurred. Please try again later.")
 
 @bot.command(name='setprefix')
 @commands.has_permissions(administrator=True)