Update app.py

app.py

@@ -5,7 +5,14 @@ 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
+import traceback
+
+try:
+    from agents import Agent, Runner, function_tool, OpenAIChatCompletionsModel
+    AGENTS_AVAILABLE = True
+except Exception:
+    print("WARNING: openai-agents package not fully functional, using fallback mode")
+    AGENTS_AVAILABLE = False
 
 # Apply nest_asyncio to allow nested event loops
 nest_asyncio.apply()
@@ -20,15 +27,9 @@ 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):
+# Direct search function that doesn't depend on the agents package
+def direct_news_search(topic, language="English", search_date=None):
+    """Search for news articles using DuckDuckGo directly"""
     # Use provided date or default to current date
     if not search_date:
         search_date = datetime.now().strftime("%Y-%m")
@@ -79,22 +80,9 @@ def get_news_articles(topic, language="English", search_date=None):
     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
-)
-
-# Fallback to direct OpenAI API if agents don't work
+# Direct OpenAI API call
 def generate_with_openai(prompt):
+    """Use OpenAI API directly to generate content"""
     try:
         response = client.chat.completions.create(
             model="gpt-4o-mini",
@@ -106,57 +94,97 @@ def generate_with_openai(prompt):
     except Exception as e:
         return f"Error with OpenAI API: {str(e)}"
 
+# Try to set up agents if available
+if AGENTS_AVAILABLE:
+    try:
+        # Define the model
+        model = OpenAIChatCompletionsModel(
+            model="gpt-4o-mini",
+            openai_client=client
+        )
+
+        # News search tool
+        @function_tool
+        def get_news_articles(topic, language="English", search_date=None):
+            return direct_news_search(topic, language, search_date)
+
+        # 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
+        )
+        
+        print("Successfully initialized agent-based workflow")
+    except Exception as e:
+        print(f"Failed to initialize agents: {e}")
+        AGENTS_AVAILABLE = False
+
 # Workflow function for Gradio
 def fetch_and_edit_news(topic, language, search_date):
+    """Process news request using agents if available, otherwise fall back to direct API calls"""
     try:
-        # Initialize new event loop
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
-        
-        # Try to use Runner.run_sync with the proper event loop context
-        try:
-            # Step 1: Run the news agent to get news
-            news_prompt = f"Get me the news about {topic} in {language} for date {search_date}"
-            news_result = Runner.run_sync(news_agent, news_prompt)
-            raw_news = news_result.final_output
-            
-            # Step 2: Run the editor agent to edit the news
-            editor_prompt = f"Please edit the following news in {language} language. Maintain the original language: \n\n{raw_news}"
-            editor_result = Runner.run_sync(editor_agent, editor_prompt)
-            edited_news = editor_result.final_output
-            
-            return edited_news
-            
-        except Exception as agent_error:
-            print(f"Agent error: {agent_error}")
-            
-            # Fallback to direct use of DuckDuckGo and OpenAI if agents fail
+        # Try agent-based approach if available
+        if AGENTS_AVAILABLE:
             try:
-                print("Falling back to direct search and API calls...")
-                # Get news directly
-                raw_news = get_news_articles(topic, language, search_date)
+                print("Attempting to use agent-based workflow...")
                 
-                if raw_news.startswith("Error") or raw_news.startswith("Could not find"):
-                    return raw_news
+                # Initialize event loop
+                loop = asyncio.new_event_loop()
+                asyncio.set_event_loop(loop)
                 
-                # Format prompt for OpenAI
-                editor_prompt = f"""
-                Please edit and reformat the following news into a cohesive, publication-ready format. 
-                Maintain the original language ({language}). 
-                Each news story should be in a separate section with a clear headline:
+                # Run the news agent
+                news_prompt = f"Get me the news about {topic} in {language} for date {search_date}"
+                news_result = Runner.run_sync(news_agent, news_prompt)
+                raw_news = news_result.final_output
                 
-                {raw_news}
-                """
+                # Run the editor agent
+                editor_prompt = f"Please edit the following news in {language} language. Maintain the original language: \n\n{raw_news}"
+                editor_result = Runner.run_sync(editor_agent, editor_prompt)
+                edited_news = editor_result.final_output
                 
-                # Call OpenAI directly
-                edited_news = generate_with_openai(editor_prompt)
+                print("Agent-based workflow completed successfully")
                 return edited_news
                 
-            except Exception as fallback_error:
-                return f"Both agent-based and fallback approaches failed.\nError details: {str(fallback_error)}"
+            except Exception as agent_error:
+                print(f"Agent-based workflow failed: {agent_error}")
+                print(traceback.format_exc())
+                print("Falling back to direct API approach...")
+                # Fall through to the direct approach
+        else:
+            print("Using direct API approach (agents not available)...")
+            
+        # Direct approach (used when agents fail or aren't available)
+        # Step 1: Get news directly
+        raw_news = direct_news_search(topic, language, search_date)
+        
+        if raw_news.startswith("Error") or raw_news.startswith("Could not find"):
+            return raw_news
+        
+        # Step 2: Edit the news with OpenAI
+        editor_prompt = f"""
+        Please edit and reformat the following news into a cohesive, publication-ready format. 
+        Maintain the original language ({language}). 
+        Each news story should be in a separate section with a clear headline:
+        
+        {raw_news}
+        """
+        
+        edited_news = generate_with_openai(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."
+        error_details = traceback.format_exc()
+        print(f"Error in workflow: {e}")
+        print(error_details)
+        return f"Error processing your request: {str(e)}\n\nPlease check that your OpenAI API key is correctly set in the repository secrets."
 
 # Create Gradio interface
 with gr.Blocks(title="Multilingual AI News Generator") as demo: