Update main.py

main.py

@@ -13,6 +13,7 @@ import time
 from datetime import datetime, timedelta
 import asyncio
 import requests
+from prompts import CODING_ASSISTANT_PROMPT, NEWS_ASSISTANT_PROMPT, generate_news_prompt
 
 app = FastAPI()
 
@@ -227,14 +228,11 @@ def analyze_news(query):
         return "Failed to fetch news data.", []
 
     # Prepare the prompt for the AI
-    prompt = f"Based on the following recent news about '{query}', Provide a well summarized and formatted answer using markdown, give importance to the latest news:\n\n"
-    for item in news_data:
-        prompt += f"Title: {item['title']}\n"
-        prompt += f"Snippet: {item['snippet']}\n"
-        prompt += f"Last Updated: {item['last_updated']}\n\n"
+    # Use the imported function to generate the prompt (now includes today's date)
+    prompt = generate_news_prompt(query, news_data)
 
     messages = [
-        {"role": "system", "content": "You are a knowledgeable Assistant capable of providing insightful answers on various topics."},
+        {"role": "system", "content": NEWS_ASSISTANT_PROMPT},
         {"role": "user", "content": prompt}
     ]