Update app.py

app.py

@@ -3,8 +3,12 @@ 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  # Assuming agents package is installed
+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
@@ -13,17 +17,10 @@ OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")  # You'll need to add this to
 # Get current date for default value
 default_date = datetime.now().strftime("%Y-%m-%d")
 
-# Configure OpenAI client to use HuggingFace or OpenAI API
-client = OpenAI(
-    api_key=OPENAI_API_KEY,
-    # If using OpenAI directly
-    # If using a different API endpoint (e.g., HF Inference API), uncomment and adjust:
-    # base_url="https://api-inference.huggingface.co/models/meta-llama/Llama-3.2-70b-instruct"
-)
-
-# Define the model - assuming the agents library supports standard OpenAI client
-from agents import OpenAIChatCompletionsModel  # Adjust import if needed
+# 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
@@ -70,14 +67,17 @@ def get_news_articles(topic, language="English", search_date=None):
     # 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}"
     
-    # 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}."
+    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(
@@ -93,32 +93,34 @@ editor_agent = Agent(
      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 event loop for this thread
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
+        # Create a new asyncio loop that works with nest_asyncio
+        loop = asyncio.get_event_loop()
         
         # Step 1: Run the news agent
-        news_result = Runner.run_sync(
-            news_agent,
-            f"Get me the news about {topic} in {language} for date {search_date}")
+        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))
         
-        raw_news = news_result.final_output
+        if raw_news.startswith("Error"):
+            return raw_news
         
         # Step 2: Pass news to editor for final review
-        editor_news_response = Runner.run_sync(
-            editor_agent,
-            f"Please edit the following news in {language} language. Maintain the original language: \n\n{raw_news}")
-        
-        edited_news = editor_news_response.final_output
+        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 the error message for debugging
-        return f"Error: {str(e)}\n\nThis could be due to API key issues or problems with the openai-agents package. Please check the logs for more details."
+        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: