commit changes to huggingface

.gitignore

@@ -0,0 +1 @@
+.venv/

app.py

@@ -0,0 +1,59 @@
+# created to retrieve prompt info for posts created to showcase a model
+# tested using python 3.10.13, streamlit 1.31.1, pandas 2.2.1, requests 2.31.0
+# if you're running on Windows python 3.10.11 should be fine
+# run from the command line "streamlit run app.py"
+import streamlit as st
+import pandas as pd
+import requests
+
+# Function to fetch images from the API
+@st.cache_data  # Use the new caching mechanism
+def fetch_images(postID):
+    try:
+        response = requests.get(f'https://civitai.com/api/v1/images?postId={postID}')
+        response.raise_for_status()  # This will raise an exception for HTTP errors
+        data = response.json()
+        items = data["items"]
+        table_data = [[item["id"], item["meta"]["prompt"], item["meta"]["negativePrompt"], item["url"]] for item in items]
+        return table_data
+    except requests.RequestException as e:
+        st.error(f"Failed to fetch images: {e}")
+        return []  # Return an empty list in case of error
+
+def display_images(df):
+    if not df.empty:
+        for _, row in df.iterrows():
+            col1, col2 = st.columns([1, 2])  # Adjust the ratio as needed
+            with col1:
+                st.image(row['url'], use_column_width=True)
+            with col2:
+                st.markdown(f"**Prompt:** {row['prompt']}")
+                st.markdown(f"**Negative Prompt:** {row['negativePrompt']}")
+
+def main():
+    st.title('Civitai Model Showcase Prompts')
+
+    st.markdown("""
+    Created to retrieve prompt info for posts created to showcase a model.
+                
+    Sample Stable Diffusion Lightning model posts - as of 2024-02-29 
+    - post id 1496616 for [DreamShaper](https://civitai.com/posts/1496616)
+    - post id 1515491 for [Juggernaut Lightning](https://civitai.com/posts/1515491)
+    """)
+
+    postID = st.text_input('Enter Post ID:', '')
+
+    if postID:
+        # Convert postID to integer if necessary
+        try:
+            postID_int = int(postID)
+            df = pd.DataFrame(fetch_images(postID_int), columns=["id", "prompt", "negativePrompt", "url"])
+            if not df.empty:
+                display_images(df)
+            else:
+                st.write("No images found for the given Post ID.")
+        except ValueError:
+            st.error("Please enter a valid Post ID.")
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,3 @@
+streamlit
+pandas
+requests