app.py

import gradio as gr
from duckduckgo_search import DDGS
from datetime import datetime
import os
import asyncio
import nest_asyncio
from openai import OpenAI  # Using standard OpenAI client
from agents import Agent, Runner, function_tool, OpenAIChatCompletionsModel  # Assuming agents package is installed

# Apply nest_asyncio to allow nested event loops
nest_asyncio.apply()

# Set up environment variables
# For Hugging Face Spaces, set these in the Settings > Repository secrets
OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")  # You'll need to add this to HF Spaces secrets

# Get current date for default value
default_date = datetime.now().strftime("%Y-%m-%d")

# Configure OpenAI client
client = OpenAI(api_key=OPENAI_API_KEY)

# Define the model
model = OpenAIChatCompletionsModel(
    model="gpt-4o-mini",  # Using GPT-4o-mini for better performance at a lower cost
    openai_client=client
)

# News search tool
@function_tool
def get_news_articles(topic, language="English", search_date=None):
    # Use provided date or default to current date
    if not search_date:
        search_date = datetime.now().strftime("%Y-%m")
    else:
        # Convert from date picker format (YYYY-MM-DD) to YYYY-MM format
        search_date = search_date[:7]  # Just get YYYY-MM portion
        
    print(f"Running DuckDuckGo news search for {topic} in {language} for date {search_date}...")
    
    # Map common languages to their search keywords
    language_keywords = {
        "English": "",  # Default, no special keyword needed
        "Hindi": "हिंदी",
        "Spanish": "español",
        "French": "français",
        "German": "deutsch",
        "Japanese": "日本語",
        "Chinese": "中文",
        "Russian": "русский",
        "Arabic": "العربية",
        "Portuguese": "português",
        "Italian": "italiano",
        "Dutch": "nederlands",
        "Korean": "한국어",
        "Turkish": "türkçe",
        "Kannada": "ಕನ್ನಡ",
        "Tamil": "தமிழ்",
        "Telugu": "తెలుగు",
        "Bengali": "বাংলা",
        "Marathi": "मराठी"
    }
    
    # Get language keyword if available
    lang_keyword = language_keywords.get(language, language)
    
    # Add language to search query if it's not English
    search_query = f"{topic} {lang_keyword} {search_date}" if language != "English" else f"{topic} {search_date}"
    
    try:
        # DuckDuckGo search
        ddg_api = DDGS()
        results = ddg_api.text(search_query, max_results=5)
        if results:
            news_results = "\n\n".join([f"Title: {result['title']}\nURL: {result['href']}\nDescription: {result['body']}" for result in results])
            return news_results
        else:
            return f"Could not find news results for {topic} in {language} for {search_date}."
    except Exception as e:
        return f"Error searching for news: {str(e)}"

# Create agents
news_agent = Agent(
    name="News Agent",
    instructions="You provide the latest news articles for a given topic using DuckDuckGo search. You can search for news in different languages when specified.",
    tools=[get_news_articles],
    model=model
)

editor_agent = Agent(
     name="Editor Assistant",
     instructions="Rewrite and give me a news article ready for publishing. Each news story should be in a separate section. Maintain the original language of the news stories. If the content is in a language other than English, edit and format in that same language.",
     model=model
)

# Helper function to safely run the agent
async def run_agent_async(agent, prompt):
    try:
        result = await Runner.arun(agent, prompt)
        return result.final_output
    except Exception as e:
        return f"Error running agent: {str(e)}"

# Workflow function for Gradio
def fetch_and_edit_news(topic, language, search_date):
    try:
        # Create a new asyncio loop that works with nest_asyncio
        loop = asyncio.get_event_loop()
        
        # Step 1: Run the news agent
        news_prompt = f"Get me the news about {topic} in {language} for date {search_date}"
        raw_news = loop.run_until_complete(run_agent_async(news_agent, news_prompt))
        
        if raw_news.startswith("Error"):
            return raw_news
        
        # Step 2: Pass news to editor for final review
        editor_prompt = f"Please edit the following news in {language} language. Maintain the original language: \n\n{raw_news}"
        edited_news = loop.run_until_complete(run_agent_async(editor_agent, editor_prompt))
        
        return edited_news
    except Exception as e:
        return f"Error: {str(e)}\n\nPlease check your OpenAI API key and ensure it has been set correctly in the repository secrets."

# Create Gradio interface
with gr.Blocks(title="Multilingual AI News Generator") as demo:
    gr.Markdown("# Multilingual AI News Generator")
    gr.Markdown("Enter a topic, select a language, and choose a date to receive curated and edited news articles")
    
    with gr.Row():
        topic_input = gr.Textbox(label="News Topic", placeholder="Enter a topic (e.g., AI, Climate Change, Sports)")
        language_dropdown = gr.Dropdown(
            choices=[
                "English", "Hindi", "Spanish", "French", "German", 
                "Japanese", "Chinese", "Russian", "Arabic", "Portuguese",
                "Italian", "Dutch", "Korean", "Turkish", "Kannada",
                "Tamil", "Telugu", "Bengali", "Marathi"
            ],
            label="Language",
            value="English"
        )
        date_picker = gr.Textbox(
            label="Search Date",
            placeholder="YYYY-MM-DD",
            value=default_date
        )
        submit_btn = gr.Button("Generate News Article")
    
    output_box = gr.Textbox(label="Generated News Article", lines=20)
    
    submit_btn.click(
        fn=fetch_and_edit_news,
        inputs=[topic_input, language_dropdown, date_picker],
        outputs=output_box
    )

# Launch the app
if __name__ == "__main__":
    demo.launch()