Update app.py

app.py

@@ -1,6 +1,27 @@
 import gradio as gr
 import requests
 import base64
+import openai
+
+# Set your OpenAI API key
+openai.api_key = "YOUR_OPENAI_API_KEY"  # Replace with your actual OpenAI API key
+
+def generate_images_from_text(text):
+    # Generate an image using OpenAI's DALL-E model or similar
+    response = openai.Image.create(
+        prompt=text,
+        n=1,
+        size="512x512"
+    )
+    
+    image_url = response['data'][0]['url']
+    
+    # Fetch the image from the URL
+    image_response = requests.get(image_url)
+    
+    # Convert the image to Base64 for embedding in HTML
+    base64_image = base64.b64encode(image_response.content).decode('utf-8')
+    return f"data:image/png;base64,{base64_image}"
 
 def generate_comic(short_story):
     sentences = short_story.split('. ')
@@ -8,25 +29,12 @@ def generate_comic(short_story):
 
     for sentence in sentences:
         if sentence.strip():
-            response = requests.post(
-                "https://api.deepai.org/api/text2img",
-                data={'text': sentence},
-                headers={'api-key': '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'}
-            )
-            # Print the response for debugging
-            print("API Response:", response.json())
-
-            image_url = response.json().get('output_url')
-            if image_url:
-                print("Generated Image URL:", image_url)  # Debugging URL
-                
-                # Convert image to base64
-                try:
-                    image_data = requests.get(image_url).content
-                    base64_image = base64.b64encode(image_data).decode('utf-8')
-                    images.append(f"data:image/png;base64,{base64_image}")
-                except Exception as e:
-                    print(f"Error fetching image: {e}")
+            # Generate image from sentence
+            try:
+                base64_image = generate_images_from_text(sentence)
+                images.append(base64_image)
+            except Exception as e:
+                print(f"Error generating image: {e}")
 
     # Create comic HTML
     comic_html = '<div style="display: flex; flex-direction: column;">'