Update app.py

app.py

@@ -1,13 +1,10 @@
 import re
 import requests
 from bs4 import BeautifulSoup
-# DuckDuckGo Search: use DDGS class
+# DuckDuckGo Search
 from duckduckgo_search import DDGS
-# Google search (unofficial scraping library)
-from googlesearch import search as google_search
 # SerpAPI client for various search engines (Google, Bing, etc.)
-# Requires SERPAPI_API_KEY environment variable or explicit key
-from serpapi import GoogleSearch as SerpApiClient
+from serpapi import GoogleSearch
 from rake_nltk import Rake
 import gradio as gr
 from transformers import pipeline
@@ -19,21 +16,18 @@ def extract_keywords(text: str) -> list[str]:
     return rake.get_ranked_phrases()[:5]
 
 # 2) Search functions
-
 def ddg_search_links(query: str, num: int = 5) -> list[str]:
     ddgs = DDGS()
     results = ddgs.text(query, max_results=num)
     return [r['href'] for r in results]
 
-
 def google_search_links(query: str, num: int = 5) -> list[str]:
-    return list(google_search(query, num_results=num, pause=2))
-
+    from googlesearch import search
+    return list(search(query, num_results=num, pause=2))
 
 def serpapi_search_links(query: str, api_key: str, engine: str = 'bing', num: int = 5) -> list[str]:
-    # engine: 'bing', 'google', 'yahoo', etc.
     params = {"engine": engine, "q": query, "api_key": api_key}
-    client = SerpApiClient(params)
+    client = GoogleSearch(params)
     data = client.get_dict()
     results = data.get('organic_results', [])
     return [r['link'] for r in results if not r.get('sponsored')][:num]
@@ -94,7 +88,6 @@ def process(prompt: str, web_enabled: bool, serpapi_key: str) -> str:
 
     # Websearch enabled: model first
     ans = model_answer(prompt)
-    # If model confident, return
     if not is_uncertain(ans):
         return ans
 
@@ -102,7 +95,6 @@ def process(prompt: str, web_enabled: bool, serpapi_key: str) -> str:
     links = []
     links += google_search_links(query)
     links += ddg_search_links(query)
-    # Bing and Google via SerpAPI
     if serpapi_key:
         links += serpapi_search_links(query, serpapi_key, engine='bing')
         links += serpapi_search_links(query, serpapi_key, engine='google')
@@ -115,9 +107,6 @@ def process(prompt: str, web_enabled: bool, serpapi_key: str) -> str:
     return summary
 
 # 8) Gradio UI
-def main(prompt: str, web_enabled: bool, serpapi_key: str) -> str:
-    return process(prompt, web_enabled, serpapi_key)
-
 with gr.Blocks() as demo:
     gr.Markdown("# Intelligente KI mit Multi-Engine-Websuche")
     with gr.Row():
@@ -128,9 +117,13 @@ with gr.Blocks() as demo:
     output = gr.Textbox(label="Antwort", lines=10)
 
     btn.click(
-        fn=main,
+        fn=process,
         inputs=[prompt_input, web_switch, serp_input],
         outputs=output
     )
 
+    gr.Spacer()
+    gr.Markdown("---")
+    gr.Markdown("*Hinweis: Suche nur bei aktivierter Websuche und nicht bei verbotenen Phrasen.*")
+
 demo.launch()