Update app.py

app.py

@@ -4,8 +4,9 @@ from datetime import datetime
 import os
 import asyncio
 import nest_asyncio
-from openai import OpenAI  # Using standard OpenAI client
+from openai import OpenAI
 import traceback
+import re
 
 try:
     from agents import Agent, Runner, function_tool, OpenAIChatCompletionsModel
@@ -18,8 +19,7 @@ except Exception:
 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
+OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 
 # Get current date for default value
 default_date = datetime.now().strftime("%Y-%m-%d")
@@ -27,14 +27,40 @@ default_date = datetime.now().strftime("%Y-%m-%d")
 # Configure OpenAI client
 client = OpenAI(api_key=OPENAI_API_KEY)
 
+# Function to make URLs clickable in Markdown
+def make_urls_clickable(text):
+    # Find URLs in the text
+    url_pattern = r'(https?://[^\s]+)'
+    
+    # Replace URLs with markdown links
+    text_with_links = re.sub(url_pattern, r'[\1](\1)', text)
+    
+    return text_with_links
+
+# Function to format article output with better readability
+def format_articles(text):
+    # Make URLs clickable
+    text = make_urls_clickable(text)
+    
+    # Add better formatting for article titles (lines starting with #)
+    text = re.sub(r'# (.+)', r'## \1', text)
+    
+    # Add horizontal rule between articles if not already present
+    if "---" not in text:
+        text = re.sub(r'\n\n(?=# )', r'\n\n---\n\n', text)
+    
+    # Enhance description and URL formatting
+    text = re.sub(r'\*\*URL\*\*: (.+)', r'**Source**: \1', text)
+    text = re.sub(r'\*\*Description\*\*:', r'**Summary**:', text)
+    
+    return text
+
 # 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")
     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}...")
@@ -73,7 +99,7 @@ def direct_news_search(topic, language="English", search_date=None):
         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])
+            news_results = "\n\n".join([f"# {result['title']}\n**URL**: {result['href']}\n**Description**: {result['body']}" for result in results])
             return news_results
         else:
             return f"Could not find news results for {topic} in {language} for {search_date}."
@@ -111,14 +137,14 @@ if AGENTS_AVAILABLE:
         # 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.",
+            instructions="You provide the latest news articles for a given topic using DuckDuckGo search. You can search for news in different languages when specified. Format each article with a # heading.",
             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.",
+             instructions="Rewrite and give me a news article ready for publishing. Each news story should be in a separate section with a clear headline (use # for headlines). 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. Keep the original URLs and mark them as **URL**: [url]. Use --- to separate articles.",
              model=model
         )
         
@@ -146,12 +172,13 @@ def fetch_and_edit_news(topic, language, search_date):
                 raw_news = news_result.final_output
                 
                 # Run the editor agent
-                editor_prompt = f"Please edit the following news in {language} language. Maintain the original language: \n\n{raw_news}"
+                editor_prompt = f"Please edit the following news in {language} language. Maintain the original language and URLs. Format with markdown, using # for headlines and --- to separate articles: \n\n{raw_news}"
                 editor_result = Runner.run_sync(editor_agent, editor_prompt)
                 edited_news = editor_result.final_output
                 
                 print("Agent-based workflow completed successfully")
-                return edited_news
+                # Format the output with better readability
+                return format_articles(edited_news)
                 
             except Exception as agent_error:
                 print(f"Agent-based workflow failed: {agent_error}")
@@ -171,14 +198,17 @@ def fetch_and_edit_news(topic, language, search_date):
         # 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:
+        Maintain the original language ({language}).
+        Format with markdown, using # for headlines and --- to separate articles.
+        Keep the original URLs and mark them as **URL**: [url].
         
         {raw_news}
         """
         
         edited_news = generate_with_openai(editor_prompt)
-        return edited_news
+        
+        # Format the output with better readability
+        return format_articles(edited_news)
             
     except Exception as e:
         error_details = traceback.format_exc()
@@ -186,42 +216,56 @@ def fetch_and_edit_news(topic, language, search_date):
         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 a theme
-with gr.Blocks(title="Multilingual AI News Generator", theme=gr.themes.Soft(primary_hue="blue")) 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"
-        )
-        # Use Date input instead of Datepicker
-        date_picker = gr.components.Textbox(
-            label="Search Date",
-            placeholder="YYYY-MM-DD",
-            value=default_date
-        )
+# Read custom CSS from file
+try:
+    with open('style.css', 'r') as f:
+        custom_css = f.read()
+except Exception as e:
+    print(f"Warning: Could not read CSS file: {e}")
+    # Fallback minimal CSS if file can't be read
+    custom_css = """
+    .container { max-width: 900px; margin: auto; }
+    .article-output { border: 1px solid #ddd; padding: 20px; }
+    """
+
+# Create Gradio interface with a custom theme
+with gr.Blocks(title="Multilingual AI News Generator", theme=gr.themes.Soft(primary_hue="blue"), css=custom_css) as demo:
+    with gr.Column(elem_classes="container"):
+        with gr.Column(elem_classes="header"):
+            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")
         
-    submit_btn = gr.Button("Generate News Article", variant="primary")
-    
-    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
-    )
+        with gr.Row(elem_classes="search-options"):
+            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", elem_classes="generate-btn")
+        
+        # Use Markdown for output to enable clickable links and better formatting
+        output_box = gr.Markdown(elem_classes="article-output")
+        
+        submit_btn.click(
+            fn=fetch_and_edit_news,
+            inputs=[topic_input, language_dropdown, date_picker],
+            outputs=output_box
+        )
 
 # Launch the app
 if __name__ == "__main__":