Update app.py

app.py

@@ -393,6 +393,60 @@ def main():
             else:
                 st.warning("Please enter an image description.")
 
+    elif search_type == "Logo Generation":
+        st.title("AI Logo Generator 🎨")
+        st.write("Generate a unique logo for your business using AI!")
+
+        # User Inputs
+        prompt = st.text_input("Enter a description for the logo:", "Make a logo for my gaming company BTZ")
+        style = st.selectbox("Select a Logo Style:", [28, 29, 30], index=0)  # Customize styles if needed
+        size = st.radio("Select Image Size:", ["1-1", "2-3", "3-4"], index=0)
+
+        def generate_logo(prompt, style, size):
+            api_url = "https://ai-logo-generator.p.rapidapi.com/aaaaaaaaaaaaaaaaaiimagegenerator/quick.php"
+            headers = {
+                "x-rapidapi-key": LOGO_API_KEY,
+                "x-rapidapi-host": "ai-logo-generator.p.rapidapi.com",
+                "Content-Type": "application/json"
+            }
+            payload = {
+                "prompt": prompt,
+                "style_id": style,
+                "size": size
+            }
+        
+            response = requests.post(api_url, json=payload, headers=headers)
+            return response.json()
+
+        # Generate button
+        if st.button("Generate Logo"):
+            with st.spinner("Generating logo... Please wait!"):
+                result = generate_logo(prompt, style, size)
+            
+                # Extracting image URLs from the JSON response
+                image_data = result.get("final_result", [])
+            
+                if image_data:
+                    st.subheader("Generated Logo Designs")
+                
+                    for index, image in enumerate(image_data):
+                        image_url = image.get("origin")  # Extracting the logo link
+                    
+                        if image_url:
+                            st.image(image_url, caption=f"Logo Design {index+1}", use_container_width=True)
+                        
+                            # Download button with a unique key
+                            st.download_button(
+                                label="Download Logo",
+                                data=requests.get(image_url).content,
+                                file_name=f"logo_design_{index+1}.webp",
+                                mime="image/webp",
+                                key=f"download_{index}"  # Unique key to prevent duplicate errors
+                            )
+                else:
+                    st.error("Failed to generate logo. No image URL found.")
+                    st.write("Response Data:", result)  # Debugging: Show full response data
+
 
     # Save chat history as PDF
     if st.button("Save Chat History as PDF"):