Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import requests | |
| from bs4 import BeautifulSoup | |
| import google.generativeai as genai | |
| def fetch_article_content(url): | |
| """Fetch article content with basic error handling""" | |
| try: | |
| headers = {'User-Agent': 'Mozilla/5.0'} | |
| response = requests.get(url, headers=headers, timeout=8) | |
| response.raise_for_status() | |
| soup = BeautifulSoup(response.text, 'html.parser') | |
| # Extract first 8 paragraphs | |
| paragraphs = soup.find_all('p')[:8] | |
| content = ' '.join([p.get_text(strip=True) for p in paragraphs]) | |
| return content[:2000] # Keep it concise | |
| except Exception as e: | |
| return f"FETCH_ERROR: {str(e)}" | |
| def generate_platform_post(article_text, api_key): | |
| """Generate post using user-provided API key""" | |
| try: | |
| # Configure Gemini with user's API key | |
| genai.configure(api_key=api_key) | |
| model = genai.GenerativeModel('gemini-1.5-pro') | |
| prompt = f""" | |
| Create a Reddit/Quora-style post from this article: | |
| {article_text} | |
| Output format: | |
| Title: [short title] | |
| Post: [clean HTML with 1 image tag + alt text] | |
| Requirements: | |
| - Title < 100 characters | |
| - Include <img> with descriptive alt text | |
| - Minimal styling | |
| - Mobile-friendly | |
| """ | |
| response = model.generate_content(prompt) # Removed invalid timeout | |
| return parse_gemini_response(response.text) | |
| except Exception as e: | |
| return ("AI ERROR", f"<p>API Error: {str(e)}</p>") | |
| def parse_gemini_response(response): | |
| """Robust response parsing – returns two values (title, content)""" | |
| try: | |
| title = response.split("Title:")[1].split("Post:")[0].strip()[:100] | |
| content = response.split("Post:")[1].strip() | |
| return title, content | |
| except: | |
| return ("Formatting Error", "<p>Failed to parse AI response</p>") | |
| def process_url(url, api_key): | |
| """Main processing pipeline with status updates""" | |
| if not api_key or len(api_key.strip()) < 30: | |
| yield ("API Key Required", "<p>Please enter your Gemini API key above</p>") | |
| return | |
| if not url.startswith("http"): | |
| yield ("Invalid URL", "<p>Please enter a valid article URL</p>") | |
| return | |
| yield ("Fetching Article...", "<p>Connecting to URL...</p>") | |
| article_text = fetch_article_content(url) | |
| if article_text.startswith("FETCH_ERROR:"): | |
| yield ("Fetch Failed", f"<p>{article_text.replace('FETCH_ERROR: ', '')}</p>") | |
| return | |
| yield ("Generating Post...", "<p>Creating content with Gemini...</p>") | |
| result = generate_platform_post(article_text, api_key) | |
| yield result # This will now return (title, content) | |
| # Create Gradio interface | |
| url_input = gr.Textbox(label="Article URL", placeholder="https://example.com/article...") | |
| api_key_input = gr.Textbox( | |
| label="Gemini API Key", | |
| placeholder="AIzaXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX", | |
| type="password" | |
| ) | |
| title_output = gr.Textbox(label="Generated Title") | |
| content_output = gr.HTML(label="Formatted Post") | |
| app = gr.Interface( | |
| fn=process_url, | |
| inputs=[url_input, api_key_input], | |
| outputs=[ | |
| gr.Textbox(label="Generated Title"), | |
| gr.HTML(label="Formatted Post") | |
| ], | |
| examples=[ | |
| ["https://google.com", "AIzaXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"] | |
| ], | |
| title="Article to Reddit/Quora Post Converter", | |
| description="Convert news articles into optimized posts with AI-generated formatting and image descriptions", | |
| allow_flagging="never", | |
| live=False, | |
| theme="default" | |
| ) | |
| if __name__ == "__main__": | |
| app.launch() |