Update app.py

app.py

@@ -20,19 +20,20 @@ def generate_image(prompt):
             raise ValueError("Model not loaded properly.")
         
         print(f"Generating image with prompt: {prompt}")
-        output = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
+        output = model(prompt=prompt, num_inference_steps=50, guidance_scale=7.5)
         print(f"Model output: {output}")
 
         if output is None:
             raise ValueError("Model returned None")
 
         if hasattr(output, 'images') and output.images:
-            print(f"Image generated")
+            print(f"Image generated successfully")
             image = output.images[0]
             buffered = BytesIO()
             image.save(buffered, format="JPEG")
             image_bytes = buffered.getvalue()
             img_str = base64.b64encode(image_bytes).decode("utf-8")
+            print("Image encoded to base64")
             print(f'img_str: {img_str}')
             return img_str, None
         else:
@@ -55,6 +56,7 @@ def inference(sentence_mapping, character_dict, selected_style):
         for paragraph_number, sentences in sentence_mapping.items():
             combined_sentence = " ".join(sentences)
             prompt = f"Make an illustration in {selected_style} style from: {combined_sentence}"
+            print(f"Generated prompt for paragraph {paragraph_number}: {prompt}")
             img_str, error = generate_image(prompt)
             if error:
                 images[paragraph_number] = f"Error: {error}"
@@ -62,6 +64,7 @@ def inference(sentence_mapping, character_dict, selected_style):
                 images[paragraph_number] = img_str
         return images
     except Exception as e:
+        print(f"An error occurred during inference: {e}")
         return {"error": str(e)}
 
 gradio_interface = gr.Interface(