Spaces:
Runtime error
Runtime error
| import os | |
| import warnings | |
| import logging | |
| # Suppress technical initialization warnings from Torch/Transformers/HF | |
| os.environ["TORCH_LOGS"] = "-all" | |
| os.environ["TORCH_CPP_LOG_LEVEL"] = "3" | |
| os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3" | |
| os.environ["HUGGINGFACE_HUB_VERBOSITY"] = "error" | |
| warnings.filterwarnings("ignore", message=".*torch.classes.*") | |
| logging.getLogger("torch").setLevel(logging.ERROR) | |
| import streamlit as st | |
| # Page Config must be first Streamlit command | |
| st.set_page_config( | |
| page_title="AutoContentForge AI", | |
| page_icon="⚒️", | |
| layout="wide", | |
| initial_sidebar_state="expanded", | |
| ) | |
| from streamlit_calendar import calendar | |
| from forge_utils import ContentExtractor, ForgeAI | |
| # Initialize ForgeAI with caching | |
| def get_forge(): | |
| return ForgeAI() | |
| forge = get_forge() | |
| # Session State for Content | |
| if 'forge_results' not in st.session_state: | |
| st.session_state.forge_results = None | |
| if 'extracted_content' not in st.session_state: | |
| st.session_state.extracted_content = None | |
| # Custom CSS for Premium Look | |
| st.markdown(""" | |
| <style> | |
| /* Global Text Color */ | |
| .stApp { | |
| color: #ffffff; | |
| } | |
| .main { | |
| background-color: #0e1117; | |
| color: #ffffff; | |
| } | |
| p, span, label, h1, h2, h3, h4, h5, h6 { | |
| color: #ffffff !important; | |
| } | |
| /* Sidebar Carbon Black */ | |
| [data-testid="stSidebar"] { | |
| background-color: #1a1c1e; | |
| border-right: 1px solid rgba(255, 255, 255, 0.05); | |
| } | |
| /* Global Button Styling - Force Red */ | |
| button, [data-testid="stBaseButton-secondary"], [data-testid="stBaseButton-primary"] { | |
| background-color: #4A4A4A !important; | |
| color: white !important; | |
| border: none !important; | |
| border-radius: 6px !important; | |
| transition: all 0.3s ease !important; | |
| } | |
| button:hover, [data-testid="stBaseButton-secondary"]:hover, [data-testid="stBaseButton-primary"]:hover { | |
| background-color: #4A4A4A !important; | |
| transform: translateY(-1px) !important; | |
| box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1) !important; | |
| color: white !important; | |
| } | |
| button:active { | |
| background-color: #991b1b !important; | |
| transform: translateY(0px) !important; | |
| } | |
| /* Fix Help Tooltips (Black text on white background) */ | |
| div[data-testid="stTooltipContent"] { | |
| background-color: #ffffff !important; | |
| color: #000000 !important; | |
| border: 1px solid #ddd !important; | |
| } | |
| div[data-testid="stTooltipContent"] p { | |
| color: #000000 !important; | |
| } | |
| /* Ensure help icons themselves are visible */ | |
| [data-testid="stWidgetLabel"] .st-at { | |
| color: #4A4A4A !important; | |
| } | |
| /* Tab and other refinements */ | |
| .stTabs [data-baseweb="tab-list"] { | |
| gap: 24px; | |
| background-color: transparent; | |
| } | |
| .stTabs [data-baseweb="tab"] { | |
| height: 50px; | |
| white-space: pre-wrap; | |
| background-color: #1a1c24; | |
| border-radius: 4px 4px 0px 0px; | |
| color: #ffffff !important; | |
| font-size: 16px; | |
| font-weight: 500; | |
| padding: 10px 20px; | |
| } | |
| .stTabs [aria-selected="true"] { | |
| background-color: #6d28d9 !important; | |
| border-bottom: 2px solid #8b5cf6 !important; | |
| } | |
| .pro-glass-card { | |
| background: rgba(255, 255, 255, 0.05); | |
| backdrop-filter: blur(10px); | |
| border: 1px solid rgba(255, 255, 255, 0.1); | |
| border-radius: 12px; | |
| padding: 20px; | |
| margin-bottom: 20px; | |
| color: #ffffff; | |
| } | |
| .platform-card { | |
| background: #1a1c24; | |
| border-left: 5px solid #6d28d9; | |
| padding: 15px; | |
| border-radius: 8px; | |
| margin-bottom: 10px; | |
| color: #ffffff; | |
| } | |
| .sidebar-pro-cta { | |
| background: linear-gradient(135deg, #6d28d9, #305CDE); | |
| color: white !important; | |
| padding: 15px; | |
| border-radius: 8px; | |
| text-align: center; | |
| margin-bottom: 20px; | |
| } | |
| .sidebar-pro-cta small, .sidebar-pro-cta strong { | |
| color: white !important; | |
| } | |
| </style> | |
| """, unsafe_allow_html=True) | |
| # Sidebar Branding & Pro CTA | |
| with st.sidebar: | |
| st.title("⚒️ AutoContentForge AI") | |
| st.caption("Automate Your Content. Forge Every Platform.") | |
| st.markdown("---") | |
| st.markdown(""" | |
| <div class="sidebar-pro-cta"> | |
| <strong>🚀 Upgrade to Pro</strong><br> | |
| <small>Self-Hosted Docker Version</small><br> | |
| <br> | |
| • Direct Social Posting<br> | |
| • Full Bulk Scheduling<br> | |
| • Video Automation<br> | |
| • Google Docs sync | |
| </div> | |
| """, unsafe_allow_html=True) | |
| if st.button("Contact for Pro Version", use_container_width=True): | |
| st.info("Email: rajue@smackcoders.com with subject 'AutoContentForge AI Pro Inquiry'") | |
| st.markdown("---") | |
| st.markdown("### Resources") | |
| st.markdown("[Documentation](#)") | |
| st.markdown("[Pricing](#)") | |
| st.markdown("[YouTube Channel](#)") | |
| # Main Navigation Tabs | |
| tab1, tab2, tab3 = st.tabs([ | |
| "📝 Single Post", | |
| "📦 Bulk Posting", | |
| "⚙️ Admin Settings" | |
| ]) | |
| # Initialize session states for the new workflow | |
| if 'master_variations' not in st.session_state: | |
| st.session_state.master_variations = None | |
| if 'selected_variation' not in st.session_state: | |
| st.session_state.selected_variation = None | |
| if 'current_hashtags' not in st.session_state: | |
| st.session_state.current_hashtags = "" | |
| # Module 1: Single Post | |
| with tab1: | |
| st.header("📝 Create Single Social Post") | |
| st.write("Fetch from a URL or use the manual editor to forge your master variations.") | |
| col_in, col_out = st.columns([1, 1.2]) | |
| with col_in: | |
| st.subheader("1. Source Content") | |
| input_mode = st.radio( | |
| "Input Mode", | |
| ["URL Fetch", "Manual Editor"], | |
| horizontal=True, | |
| help="Choose 'URL Fetch' to extract content from a website, or 'Manual Editor' to paste your own text directly." | |
| ) | |
| if input_mode == "URL Fetch": | |
| source_url = st.text_input( | |
| "Enter Content URL", | |
| placeholder="https://example.com/blog-post", | |
| help="Paste the full link (including https://) of the article or blog post you want to repurpose." | |
| ) | |
| if st.button("🚀 Fetch & Analyze"): | |
| if not source_url or "://" not in source_url: | |
| st.error("Please enter a valid URL (including https://)") | |
| else: | |
| with st.spinner("Forging 3 master variations..."): | |
| extracted = ContentExtractor.from_url(source_url) | |
| if "error" in extracted: | |
| st.error(extracted["error"]) | |
| else: | |
| st.session_state.extracted_content = extracted | |
| st.session_state.master_variations = forge.generate_master_variations(extracted['text']) | |
| st.session_state.current_hashtags = forge.extract_hashtags(extracted['text']) | |
| st.session_state.selected_variation = None # Reset selection | |
| st.success("Successfully generated 3 variations!") | |
| else: | |
| pasted_title = st.text_input("Post Title (Optional)", help="Give your post a title for easier tracking.") | |
| pasted_text = st.text_area( | |
| "Master Content / Article Text", | |
| height=200, | |
| help="Paste the core content you want to turn into social posts. Aim for at least 50 words for the best AI results." | |
| ) | |
| if st.button("✨ Generate Variations"): | |
| if len(pasted_text) < 50: | |
| st.warning("Please provide more content for better AI results.") | |
| else: | |
| with st.spinner("Processing..."): | |
| st.session_state.extracted_content = {"title": pasted_title or "Manual Input", "text": pasted_text} | |
| st.session_state.master_variations = forge.generate_master_variations(pasted_text) | |
| st.session_state.current_hashtags = forge.extract_hashtags(pasted_text) | |
| st.session_state.selected_variation = None | |
| st.success("Variations generated!") | |
| if st.session_state.master_variations: | |
| st.markdown("---") | |
| st.subheader("2. Choose Master Variation") | |
| st.info("💡 **Tip:** Each variation has a different tone. Expand them to see which one fits your brand voice best.") | |
| for var in st.session_state.master_variations: | |
| with st.expander(f"🔹 {var['type']}", expanded=(st.session_state.selected_variation == var['id'])): | |
| st.write(var['text']) | |
| if st.button(f"Select {var['type']}", key=f"sel_{var['id']}", help=f"Choose the {var['type']} to be adapted for your social networks."): | |
| st.session_state.selected_variation = var['id'] | |
| st.rerun() | |
| with col_out: | |
| st.subheader("3. Preview & Adapt") | |
| if not st.session_state.selected_variation: | |
| st.info("Select a master variation on the left to see platform-specific previews.") | |
| else: | |
| selected_var_obj = next(v for v in st.session_state.master_variations if v['id'] == st.session_state.selected_variation) | |
| platforms = st.multiselect( | |
| "Select Social Networks", | |
| ["LinkedIn", "Twitter/X", "Instagram", "Facebook", "YouTube", "Bluesky"], | |
| default=["LinkedIn", "Twitter/X"], | |
| help="Choose which platforms you want to generate previews for. The AI will format your selected variation for each." | |
| ) | |
| if st.session_state.extracted_content.get("images"): | |
| st.image(st.session_state.extracted_content['images'][0], caption="Extracted Media", width=300) | |
| else: | |
| st.write("📸 *No media found. You can add media in the Pro version or manual editor.*") | |
| st.markdown("---") | |
| for platform in platforms: | |
| adapted_text = forge.adapt_for_platform(selected_var_obj['text'], platform, st.session_state.current_hashtags) | |
| with st.container(): | |
| st.markdown(f""" | |
| <div class="platform-card"> | |
| <strong>{platform} Preview</strong><br> | |
| <p style='font-size: 0.9em; color: #ccc;'> | |
| {adapted_text if isinstance(adapted_text, str) else adapted_text.get('title')} | |
| </p> | |
| </div> | |
| """, unsafe_allow_html=True) | |
| if isinstance(adapted_text, dict) and platform == "YouTube": | |
| st.caption(f"Description: {adapted_text.get('description')}") | |
| if st.button("📅 Post / Schedule Now", type="primary"): | |
| st.success("✅ Successfully queued! (Flow simulation)") | |
| st.balloons() | |
| st.info("🔒 In Pro: This would instantly publish or sync to your drag-and-drop calendar.") | |
| # Module 2: Bulk Posting | |
| with tab2: | |
| st.header("📦 Bulk URL Posting") | |
| st.write("Forge content from multiple URLs at once. Perfect for content migration or high-volume publishing.") | |
| bulk_urls = st.text_area( | |
| "List of URLs (one per line)", | |
| height=250, | |
| placeholder="https://site.com/post-1\nhttps://site.com/post-2", | |
| help="Enter a list of blog post links. In the Pro version, the AI will process all of them in one batch." | |
| ) | |
| st.markdown(""" | |
| <div class="pro-glass-card"> | |
| <strong>Pro Feature Highlights:</strong><br> | |
| • Automatic batch processing of sitemaps<br> | |
| • Intelligent image matching for bulk lists<br> | |
| • Scheduled intervals (e.g., 1 post every 2 hours)<br> | |
| • CSV / JSON Export for external schedulers | |
| </div> | |
| """, unsafe_allow_html=True) | |
| st.button("🚀 Process Bulk List", disabled=True, help="This feature is exclusive to the Pro version.") | |
| st.markdown("---") | |
| if st.button("Unlock Bulk & Automation", key="bulk_pro", help="Contact us to upgrade to the full self-hosted version."): | |
| st.info("Email: rajue@smackcoders.com with subject 'AutoContentForge AI Pro Inquiry'") | |
| # Module 3: Admin Settings | |
| with tab3: | |
| st.header("⚙️ Admin Settings") | |
| st.write("Authorize and manage your social network connections.") | |
| networks = [ | |
| {"name": "LinkedIn", "icon": "👔"}, | |
| {"name": "Twitter/X", "icon": "🐦"}, | |
| {"name": "Instagram", "icon": "📸"}, | |
| {"name": "Facebook", "icon": "👥"}, | |
| {"name": "YouTube", "icon": "🎥"}, | |
| {"name": "Bluesky", "icon": "🦋"} | |
| ] | |
| cols = st.columns(3) | |
| for i, net in enumerate(networks): | |
| with cols[i % 3]: | |
| st.markdown(f"### {net['icon']} {net['name']}") | |
| st.button(f"Authorize {net['name']}", key=f"auth_btn_{net['name']}", help=f"Connect your {net['name']} account via OAuth for direct posting.") | |
| st.caption(f"Status: Not Connected") | |
| st.markdown("---") | |
| st.subheader("Global Settings") | |
| st.checkbox("Auto-add smart hashtags to all posts", value=True, help="Automatically suggest and append hashtags based on content analysis.") | |
| st.checkbox("Notify me on successful publishing", value=True, help="Send a notification once a post is live on the selected platform.") | |
| st.selectbox("Default AI Tone", ["Professional", "Conversational", "Exciting", "Detailed"], help="Select the default voice for your AI-generated master variations.") | |
| if st.button("Save Settings", help="Save your preferences for future sessions."): | |
| st.success("Settings saved locally! (Simulation)") | |