Update app.py

app.py

@@ -12,7 +12,7 @@ from PIL import Image
 from io import BytesIO
 from fpdf import FPDF
 import base64
-from duckduckgo_search import ddg_images
+from duckduckgo_search import DDGS
 
 # Load environment variables
 load_dotenv()
@@ -51,13 +51,14 @@ def call_llm(messages, model="deepseek/deepseek-chat-v3-0324:free", max_tokens=3
                         except json.JSONDecodeError:
                             pass
 
+
+
 def get_sources(topic, domains=None):
     query = topic
     if domains:
         domain_filters = [d.strip() for d in domains.split(",") if d.strip()]
         query += " site:" + " OR site:".join(domain_filters)
 
-    # Fetch search results from Tavily
     response = tavily.search(query=query, search_depth="advanced", max_results=10)
     results = []
 
@@ -67,8 +68,11 @@ def get_sources(topic, domains=None):
         # Fallback to DuckDuckGo image search if Tavily doesn't return an image
         if not image_url:
             try:
-                ddg_results = ddg_images(r["title"], max_results=1)
-                image_url = ddg_results[0]["image"] if ddg_results else None
+                with DDGS() as ddgs:
+                    ddgs_images_gen = ddgs.images(r["title"])
+                    image_result = next(ddgs_images_gen, None)
+                    if image_result:
+                        image_url = image_result.get("image")
             except Exception as e:
                 print(f"DuckDuckGo image search failed for '{r['title']}': {e}")
                 image_url = None