debugger

app.py

@@ -1,13 +1,16 @@
 import os
 import threading
 import requests
+import logging
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import PlainTextResponse, JSONResponse
 from FLUX import generate_image
 from VoiceReply import generate_voice_reply
-
 from llm import generate_llm
 
+# Configure logging for debugging
+logging.basicConfig(level=logging.DEBUG, format="%(asctime)s [%(levelname)s] %(message)s")
+
 GREEN_API_URL = os.getenv("GREEN_API_URL")
 GREEN_API_MEDIA_URL = os.getenv("GREEN_API_MEDIA_URL", "https://api.green-api.com")
 GREEN_API_TOKEN = os.getenv("GREEN_API_TOKEN")
@@ -23,11 +26,7 @@ if not all([GREEN_API_URL, GREEN_API_TOKEN, GREEN_API_ID_INSTANCE, WEBHOOK_AUTH_
 app = FastAPI()
 
 def send_message(message_id, to_number, message, retries=3):
-    if to_number.endswith('@g.us'):
-        chat_id = to_number
-    else:
-        chat_id = to_number
-
+    chat_id = to_number if to_number.endswith('@g.us') else to_number
     url = f"{GREEN_API_URL}/waInstance{GREEN_API_ID_INSTANCE}/sendMessage/{GREEN_API_TOKEN}"
     payload = {
         "chatId": chat_id,
@@ -46,11 +45,7 @@ def send_message(message_id, to_number, message, retries=3):
             return {"error": str(e)}
 
 def send_image(message_id, to_number, image_path, retries=3):
-    if to_number.endswith('@g.us'):
-        chat_id = to_number
-    else:
-        chat_id = to_number
-
+    chat_id = to_number if to_number.endswith('@g.us') else to_number
     url = f"{GREEN_API_MEDIA_URL}/waInstance{GREEN_API_ID_INSTANCE}/sendFileByUpload/{GREEN_API_TOKEN}"
     payload = {'chatId': chat_id, 'caption': 'Here you go!', 'quotedMessageId': message_id}
     files = [('file', ('image.jpg', open(image_path, 'rb'), 'image/jpeg'))]
@@ -69,14 +64,11 @@ def send_audio(message_id, to_number, audio_path, retries=3):
     """
     Send an audio file using the Green API similar to send_image.
     """
-    print("DEBUG: Entering send_audio")
-    if to_number.endswith('@g.us'):
-        chat_id = to_number
-    else:
-        chat_id = to_number
+    logging.debug("Entering send_audio")
+    chat_id = to_number if to_number.endswith('@g.us') else to_number
 
     if not os.path.exists(audio_path):
-        print(f"DEBUG: Audio file does not exist: {audio_path}")
+        logging.debug(f"Audio file does not exist: {audio_path}")
 
     url = f"{GREEN_API_MEDIA_URL}/waInstance{GREEN_API_ID_INSTANCE}/sendFileByUpload/{GREEN_API_TOKEN}"
     payload = {'chatId': chat_id, 'caption': 'Here is your voice reply!', 'quotedMessageId': message_id}
@@ -85,18 +77,18 @@ def send_audio(message_id, to_number, audio_path, retries=3):
             files = [('file', ('audio.mp3', audio_file, 'audio/mpeg'))]
             for attempt in range(retries):
                 try:
-                    print(f"DEBUG: Attempt {attempt + 1} to send audio")
+                    logging.debug(f"Attempt {attempt + 1} to send audio")
                     response = requests.post(url, data=payload, files=files)
-                    print("DEBUG: Response from send_audio:", response.status_code, response.text)
+                    logging.debug(f"Response from send_audio: {response.status_code} {response.text}")
                     response.raise_for_status()
                     return response.json()
                 except requests.RequestException as e:
-                    print(f"DEBUG: Exception on attempt {attempt + 1} in send_audio: {e}")
+                    logging.debug(f"Exception on attempt {attempt + 1} in send_audio: {e}")
                     if attempt < retries - 1:
                         continue
                     return {"error": str(e)}
     except Exception as e:
-        print("DEBUG: Failed to open audio file:", e)
+        logging.debug(f"Failed to open audio file: {e}")
         return {"error": str(e)}
 
 def response_text(message_id, chat_id, prompt):
@@ -107,22 +99,22 @@ def response_text(message_id, chat_id, prompt):
         send_message(message_id, chat_id, "There was an error processing your request.")
 
 def response_audio(message_id, chat_id, prompt):
-    print("DEBUG: Entering response_audio with prompt:", prompt)
+    logging.debug(f"Entering response_audio with prompt: {prompt}")
     try:
         result = generate_voice_reply(prompt, model="openai-audio", voice="coral", audio_dir=audio_dir)
-        print("DEBUG: Result from generate_voice_reply:", result)
+        logging.debug(f"Result from generate_voice_reply: {result}")
         if result:
             audio_file_path, audio_data = result
-            print("DEBUG: Audio file path generated:", audio_file_path)
+            logging.debug(f"Audio file path generated: {audio_file_path}")
             send_result = send_audio(message_id, chat_id, audio_file_path)
-            print("DEBUG: Result from send_audio:", send_result)
+            logging.debug(f"Result from send_audio: {send_result}")
             os.remove(audio_file_path)  # Clean up the file after sending
-            print("DEBUG: Removed audio file:", audio_file_path)
+            logging.debug(f"Removed audio file: {audio_file_path}")
         else:
-            print("DEBUG: generate_voice_reply returned None, falling back to response_text")
+            logging.debug("generate_voice_reply returned None, falling back to response_text")
             response_text(message_id, chat_id, prompt=prompt)
     except Exception as e:
-        print("DEBUG: Exception in response_audio:", e)
+        logging.debug(f"Exception in response_audio: {e}")
         send_message(message_id, chat_id, "There was an error generating the audio. Please try again later.")
 
 def handle_image_generation(message_id, chat_id, prompt):
@@ -130,7 +122,11 @@ def handle_image_generation(message_id, chat_id, prompt):
         image, image_path, returned_prompt, image_url = generate_image(prompt, message_id, message_id, image_dir)
         if image:
             send_image(message_id, chat_id, image_path)
-            send_message(message_id, chat_id, f"Image generated successfully! You can view it here: {image_url}. \nPrompt: > _{returned_prompt}_")
+            send_message(
+                message_id,
+                chat_id,
+                f"Image generated successfully! You can view it here: {image_url}.\n>{chr(8203)} _{returned_prompt}_"
+            )
         else:
             send_message(message_id, chat_id, "Failed to generate image. Please try again later.")
     except Exception as e:
@@ -145,20 +141,16 @@ async def whatsapp_webhook(request: Request):
     auth_header = request.headers.get('Authorization', '').strip()
     if auth_header != f"Bearer {WEBHOOK_AUTH_TOKEN}":
         raise HTTPException(status_code=403, detail="Unauthorized")
-
     try:
         data = await request.json()
     except Exception:
         return JSONResponse(content={"error": "Invalid JSON"}, status_code=400)
-
     if data.get('typeWebhook') != 'incomingMessageReceived':
         return {"success": True}
-
     try:
         chat_id = data['senderData']['chatId']
         message_id = data['idMessage']
         message_data = data.get('messageData', {})
-
         if 'textMessageData' in message_data:
             body = message_data['textMessageData']['textMessage'].strip()
         elif 'extendedTextMessageData' in message_data:
@@ -167,7 +159,6 @@ async def whatsapp_webhook(request: Request):
             return {"success": True}
     except KeyError as e:
         return JSONResponse(content={"error": f"Missing key in data: {e}"}, status_code=200)
-
     if body.lower().startswith('/imagine'):
         prompt = body.replace('/imagine', '').strip()
         if not prompt:
@@ -177,12 +168,11 @@ async def whatsapp_webhook(request: Request):
             threading.Thread(target=handle_image_generation, args=(message_id, chat_id, prompt)).start()
     else:
         threading.Thread(target=response_text, args=(message_id, chat_id, body)).start()
-
     return {"success": True}
 
 def main():
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=PORT, debug=True)
+    uvicorn.run(app, host="0.0.0.0", port=PORT)
 
 if __name__ == '__main__':
     main()