Update test_openai.py

test_openai.py

@@ -1,23 +1,30 @@
-import requests
-from PIL import Image
+import openai
 import streamlit as st
+from PIL import Image
+import requests
+from io import BytesIO
 
-# Function to generate image from Hugging Face API (or other providers like DeepAI)
+# Function to generate image from OpenAI API
 def generate_image_from_text(text):
-    response = requests.post(
-        "https://api-inference.huggingface.co/models/dalle-mini", 
-        headers={"Authorization": f"Bearer {st.secrets['HF_API_KEY']}"},
-        json={"inputs": text}
-    )
-    if response.status_code == 200:
-        image_url = response.json()[0]['image']
+    try:
+        response = openai.Image.create(
+            prompt=text,  # The text prompt for the image
+            n=1,  # Number of images to generate
+            size="1024x1024"  # Size of the image
+        )
+        # Extracting the image URL from the response
+        image_url = response['data'][0]['url']
         return image_url
-    else:
+    except Exception as e:
+        st.error(f"Error generating image: {e}")
         return None
 
-# Test Image Generation (on button click or any other trigger)
+# Streamlit UI for Image Generation
 if __name__ == "__main__":
-    st.title("Test Image Generation API")
+    st.title("Test OpenAI Image Generation API")
+
+    # Load OpenAI API key securely
+    openai.api_key = st.secrets["OPENAI_API_KEY"]
 
     user_input = st.text_input("Enter text for image generation:")
 
@@ -25,7 +32,13 @@ if __name__ == "__main__":
         if user_input:
             image_url = generate_image_from_text(user_input)
             if image_url:
-                st.image(image_url, caption="Generated Image")
+                # Fetch the image from the URL and display it
+                response = requests.get(image_url)
+                if response.status_code == 200:
+                    img = Image.open(BytesIO(response.content))
+                    st.image(img, caption="Generated Image")
+                else:
+                    st.error("Failed to fetch image.")
             else:
                 st.error("Image generation failed.")
         else: