Update app.py

app.py

@@ -2,45 +2,49 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import requests
 
-# Funktion, um Kontext von Wikipedia zu holen
-def get_wikipedia_summary(query):
-    try:
-        # Wir nutzen die öffentliche Wikipedia API
-        response = requests.get(
-            "https://de.wikipedia.org/w/api.php",
-            params={
-                "action": "query",
-                "format": "json",
-                "list": "search",
-                "srsearch": query,
-                "srlimit": 1
-            }
-        ).json()
-        
-        if not response["query"]["search"]:
-            return None
-            
-        page_id = response["query"]["search"][0]["pageid"]
-        
-        # Details zur Seite holen
-        details = requests.get(
-            "https://de.wikipedia.org/w/api.php",
-            params={
-                "action": "query",
-                "format": "json",
-                "prop": "extracts",
-                "pageids": page_id,
-                "explaintext": True,
-                "exintro": True,
-                "exsentences": 7 # Nur die ersten 7 Sätze
-            }
-        ).json()
-        
-        page = details["query"]["pages"][str(page_id)]
-        return page["extract"]
-    except Exception as e:
+def wiki_info(frage: str) -> str:
+    search_url = "https://de.wikipedia.org/w/api.php"
+    search_params = {
+        "action": "query",
+        "list": "search",
+        "srsearch": frage,
+        "format": "json"
+    }
+
+    r = requests.get(search_url, params=search_params)
+    data = r.json()
+
+    suchergebnisse = data.get("query", {}).get("search", [])
+    if not suchergebnisse:
+        return None
+
+    pageid = suchergebnisse[0]["pageid"]
+
+    # 2. Volltext laden
+    page_url = "https://de.wikipedia.org/w/api.php"
+    page_params = {
+        "action": "query",
+        "prop": "extracts",
+        "pageids": pageid,
+        "explaintext": 1,
+        "format": "json"
+    }
+
+    r2 = requests.get(page_url, params=page_params)
+    page_data = r2.json()
+
+    pages = page_data.get("query", {}).get("pages", {})
+    page = pages.get(str(pageid), {})
+
+    text = page.get("extract")
+    if not text:
         return None
 
+    saetze = re.split(r'(?<=[.!?]) +', text)
+    gekuerzt = " ".join(saetze[:10])
+
+    return gekuerzt
+
 def respond(
     message,
     history: list[dict[str, str]],