Update app.py

app.py

@@ -10,26 +10,33 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 from peft import PeftModel
 import openai
 from openai import OpenAI
+import logging
 
 
 
 openai.api_key = os.environ.get("OPENAI_KEY")
 
+# Set up logging configuration
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
+
+# Example usage of logging in your function
 def generate_image(text):
     try:
+        logging.debug("Generating image with prompt: %s", text)
         response = openai.images.generate(
             model="dall-e-3",
-            prompt="Create an illustration that accurately depicts the character and the setting of a story:"+text,
+            prompt="Create an illustration that accurately depicts the character and the setting of a story:" + text,
             n=1,
             size="1024x1024"
         )
+        image_url = response.data[0].url
+        logging.info("Image generated successfully: %s", image_url)
+        return image_url
     except Exception as error:
-        print(str(error))
+        logging.error("Failed to generate image: %s", str(error))
         raise gr.Error("An error occurred while generating the image. Please check your API key and try again.")
 
     
-    return response.data[0].url
-    
 
 # Constants
 MAX_MAX_NEW_TOKENS = 2048
@@ -99,6 +106,8 @@ def generate(
     top_k: int = 20,
     repetition_penalty: float = 1.0,
 ) -> Iterator[str]:
+    logging.debug("Starting chat generation with message: %s", message)
+
     conversation = []
     for user, assistant in chat_history:
         conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])