add alternate source for dimensions

app.py

@@ -10,7 +10,7 @@ def fetch_images(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"].get("negativePrompt", "N/A"), item["meta"]["Model"],item["meta"]["sampler"], item["meta"]["steps"], item["meta"]["cfgScale"], item["meta"].get("Size", "missing"), item["url"]] for item in items]
+        table_data = [[item["id"], item["meta"]["prompt"], item["meta"].get("negativePrompt", "N/A"), item["meta"]["Model"],item["meta"]["sampler"], item["meta"]["steps"], item["meta"]["cfgScale"], item["meta"].get("Size", "missing"), item["meta"].get("width", "missing"), item["meta"].get("height", "missing"), item["url"]] for item in items]
         return table_data
     except requests.RequestException as e:
         st.error(f"Failed to fetch images: {e}")
@@ -29,7 +29,7 @@ def display_images(df):
                 st.markdown(f"**Model:** {row['Model']}")
                 st.markdown(f"**Sampler:** {row['sampler']}    |    **Steps:** {row['steps']}")
                 st.markdown(f"**Config Scale:** {row['cfgScale']}")
-                st.markdown(f"**Image Size:** {row['Size']} (W x H)")
+                st.markdown(f"**Image Size:** {row['Size']} (W x H)   OR  {row['width']} x {row['height']} (W x H))")
 def main():
     st.title('Civitai Posts - Prompt Summary')
 
@@ -52,7 +52,7 @@ def main():
         # Convert postID to integer if necessary
         try:
             postID_int = int(postID_input)
-            df = pd.DataFrame(fetch_images(postID_int), columns=["id", "prompt", "negativePrompt", "Model", "sampler", "steps", "cfgScale", "Size", "url"])
+            df = pd.DataFrame(fetch_images(postID_int), columns=["id", "prompt", "negativePrompt", "Model", "sampler", "steps", "cfgScale", "Size", "width", "height", "url"])
             if not df.empty:
                 display_images(df)
             else: