Update utils/ui_helpers.py

utils/ui_helpers.py

@@ -1,67 +1,22 @@
 import streamlit as st
-import pandas as pd
 
-def display_results(data, content_type):
-    """Display extraction results in Streamlit UI"""
-    st.success("✅ Content extracted successfully!")
-    
-    # Metadata section
-    with st.expander("📊 Metadata", expanded=True):
-        col1, col2, col3 = st.columns(3)
-        col1.metric("Status Code", data["metadata"]["status_code"])
-        col2.metric("Content Type", content_type)
-        col3.metric("Timestamp", data["metadata"]["timestamp"])
-    
-    # Content sections based on what was extracted
-    if "article" in data["content"]:
-        with st.expander("📝 Article Content", expanded=True):
-            st.subheader(data["content"]["article"]["title"])
-            st.write(data["content"]["article"]["content"])
-    
-    if "text" in data["content"]:
-        with st.expander("📄 Text Content"):
-            for i, text in enumerate(data["content"]["text"][:10]):  # Show first 10 paragraphs
-                st.write(text)
-                if i < len(data["content"]["text"][:10]) - 1:
-                    st.divider()
-    
-    if "images" in data["content"] and data["content"]["images"]:
-        with st.expander("🖼️ Images"):
-            for img in data["content"]["images"][:5]:  # Show first 5 images
-                st.image(img["src"], caption=img["alt"], use_column_width=True)
-    
-    if "links" in data["content"] and data["content"]["links"]:
-        with st.expander("🔗 Links"):
-            links_df = pd.DataFrame(data["content"]["links"])
-            st.dataframe(links_df[["text", "href"]].head(10))  # Show first 10 links
-    
-    if "custom" in data["content"] and data["content"]["custom"]:
-        with st.expander("🎯 Custom Elements"):
-            for element in data["content"]["custom"]:
-                st.write(f"**Selector**: `{element['selector']}`")
-                st.write(f"**Text**: {element['text']}")
-                st.divider()
+def styled_container(title, content):
+    return st.markdown(f"""
+    <div class="card">
+        <h3>{title}</h3>
+        {content}
+    </div>
+    """, unsafe_allow_html=True)
 
-def render_content(content_type, data):
-    """Render content in a type-specific way"""
-    if content_type == "quotes":
-        render_quotes(data)
-    elif content_type == "newsletter":
-        render_newsletter(data)
-
-def render_quotes(data):
-    """Specialized rendering for quotes"""
-    if "custom" in data["content"]:
-        st.subheader("💬 Extracted Quotes")
-        for element in data["content"]["custom"]:
-            if "quote" in element["selector"]:
-                st.info(f'"{element["text"]}"')
-                st.divider()
-
-def render_newsletter(data):
-    """Specialized rendering for news content"""
-    if "custom" in data["content"]:
-        st.subheader("📰 News Headlines")
-        for element in data["content"]["custom"]:
-            if any(sel in element["selector"] for sel in [".title", ".storylink"]):
-                st.write(f"• {element['text']}")
+def render_footer():
+    st.divider()
+    st.markdown("""
+    <div style="text-align: center; padding: 1.5rem; color: #666;">
+        <p>Built with ❤️ using Python, FastAPI, and Streamlit</p>
+        <p>
+            <a href="https://github.com/yourusername/webtapi" target="_blank">GitHub</a> | 
+            <a href="https://huggingface.co/spaces" target="_blank">Hugging Face</a> | 
+            AGPL-3.0 License
+        </p>
+    </div>
+    """, unsafe_allow_html=True)