Update app.py

app.py

@@ -93,32 +93,68 @@ editor_agent = Agent(
      model=model
 )
 
-# Helper function to safely run the agent
-async def run_agent_async(agent, prompt):
+# Fallback to direct OpenAI API if agents don't work
+def generate_with_openai(prompt):
     try:
-        result = await Runner.arun(agent, prompt)
-        return result.final_output
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.7,
+            max_tokens=2000
+        )
+        return response.choices[0].message.content
     except Exception as e:
-        return f"Error running agent: {str(e)}"
+        return f"Error with OpenAI API: {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))
+        # Initialize new event loop
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
         
-        return edited_news
+        # 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:
+                print("Falling back to direct search and API calls...")
+                # Get news directly
+                raw_news = get_news_articles(topic, language, search_date)
+                
+                if raw_news.startswith("Error") or raw_news.startswith("Could not find"):
+                    return raw_news
+                
+                # 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:
+                
+                {raw_news}
+                """
+                
+                # Call OpenAI directly
+                edited_news = generate_with_openai(editor_prompt)
+                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 e:
         return f"Error: {str(e)}\n\nPlease check your OpenAI API key and ensure it has been set correctly in the repository secrets."