Upload 4 files

app.py

@@ -0,0 +1,34 @@
+import streamlit as st
+from llm import Model
+from image_engine import ImageGenerator
+import io
+
+# Instantiate the models
+gemini_model = Model()
+image_generator = ImageGenerator()
+
+# Streamlit interface
+st.title("AI Arist🎨")
+art_idea = st.text_input("Enter your drawing idea:")
+
+if st.button("Generate Art"):
+    if art_idea:
+        with st.spinner("Enhancing your idea..."):
+            enhanced_idea = gemini_model.enhance_idea(art_idea)
+            st.write("Enhanced Idea:", enhanced_idea)
+        with st.spinner("Geneating an image..."):
+            image = image_generator.generate_image(enhanced_idea)
+            st.image(image, caption="Generated Art")
+            
+            # Add a download button for the generated image
+            img_byte_arr = io.BytesIO()
+            image.save(img_byte_arr, format='PNG')
+            img_byte_arr = img_byte_arr.getvalue()
+            st.download_button(
+                label="Download",
+                data=img_byte_arr,
+                file_name="generated_art.png",
+                mime="image/png"
+            )
+    else:
+        st.error("Please enter an art idea!")

image_engine.py

@@ -0,0 +1,35 @@
+from PIL import Image
+import requests
+import io
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+
+# Define the ImageGenerator class for Hugging Face API
+class ImageGenerator:
+    def __init__(self) -> None:
+        self.api_url = "https://api-inference.huggingface.co/models/sd-community/sdxl-flash"
+        self.headers = {"Authorization": f"Bearer {os.getenv('HUGGING_FACE')}"}
+    
+    def generate_image(self, prompt):
+        payload = {
+            "inputs": prompt,
+            # "negative_prompt": "ugly, distorted, low quality"
+        }
+        response = requests.post(self.api_url, headers=self.headers, json=payload)
+        
+        # Debugging: Print response status and content type
+        print(f"Response Status Code: {response.status_code}")
+        print(f"Response Content Type: {response.headers.get('Content-Type')}")
+
+        # Check for valid response
+        if response.status_code == 200 and response.headers.get('Content-Type').startswith('image'):
+            image_bytes = response.content
+            image = Image.open(io.BytesIO(image_bytes))
+            return image
+        else:
+            # Print the error message if not a valid image
+            print("Error: Invalid response received from API")
+            print(response.text)  # Print the text content of the response for debugging
+            raise Exception("Failed to generate image from API")

llm.py

@@ -0,0 +1,26 @@
+import google.generativeai as genai
+from dotenv import load_dotenv
+import os
+
+    
+# Load environment variables
+load_dotenv()
+os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
+genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
+
+# Define the GeminiModel class for Gemini LLM
+class Model:
+    def __init__(self) -> None:
+        self.model = genai.GenerativeModel('gemini-1.5-flash-latest')
+    
+    def enhance_idea(self, idea):
+        prompt = """Give me the Pencil art prompt with this idea using this key words [Contrast: Utilize varying shades of gray to create depth and dimension.
+                    Texture: Capture the texture of different surfaces like wood, fabric, or skin.
+                    Detail: Focus on intricate details to enhance realism.
+                    Shading: Use different pencil pressures to create smooth gradients and sharp contrasts.
+                    Light and Shadow: Create the illusion of light sources and how they interact with objects.
+                    Composition: Arrange elements within the drawing to create a visually appealing layout.
+                    Strokes: Experiment with different pencil strokes (e.g., hatching, cross-hatching, stippling) for texture and shading.
+                    Negative Space: Use empty areas to define shapes and create balance.], \n Idea: """ + idea + "\n important note: your response must only contain the idea of classic B&W pencil art, without unnecessary elements like ('Enhanced Art Idea:' and markdown format like '##', '*')"
+        response = self.model.generate_content([prompt])
+        return response.text

requirements.txt

@@ -0,0 +1,5 @@
+streamlit
+google-generativeai
+python-dotenv
+requests 
+pillow