Update test_openai.py

test_openai.py

@@ -1,20 +1,32 @@
-import openai
+import requests
+from PIL import Image
+import streamlit as st
 
-# Set your OpenAI API key
-openai.api_key = "YOUR_OPENAI_API_KEY"  # Replace with your OpenAI API key
+# Function to generate image from Hugging Face API (or other providers like DeepAI)
+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']
+        return image_url
+    else:
+        return None
 
-# Define a quote
-random_text = "The only way to do great work is to love what you do."
+# Test Image Generation (on button click or any other trigger)
+if __name__ == "__main__":
+    st.title("Test Image Generation API")
 
-# Call OpenAI's DALL·E model to generate an image from the quote
-response = openai.Image.create(
-    prompt=random_text,  # Use the random quote as the prompt
-    n=1,  # Generate 1 image
-    size="1024x1024"  # Image size
-)
+    user_input = st.text_input("Enter text for image generation:")
 
-# Extract the image URL from the response
-image_url = response['data'][0]['url']
-
-# Print the image URL
-print(f"Image URL: {image_url}")
+    if st.button("Generate Image"):
+        if user_input:
+            image_url = generate_image_from_text(user_input)
+            if image_url:
+                st.image(image_url, caption="Generated Image")
+            else:
+                st.error("Image generation failed.")
+        else:
+            st.error("Please enter text to generate an image.")