Update app.py

app.py

@@ -14,7 +14,7 @@ custom_css = """
 #md {
     height: 400px;  
     font-size: 30px;
-    background: #202020;
+    background: #121212;
     padding: 20px;
     color: white;
     border: 1 px solid white;
@@ -34,17 +34,7 @@ def websearch(prompt):
     headers = {
         "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
     }
-    #url = f"https://www.googleapis.com/customsearch/v1?key={google_api_key}&cx={cx}&q={prompt}"
-    url = f"https://www.google.com/search?q={prompt}"
-
-    response = requests.get(url, headers=headers)
-    soup = BeautifulSoup(response.content, 'html.parser')
-    response_text = soup.find('body')
-    #prompt = f"{search_term}\n use this result from a google search to answer the question \n {response_text.text}"
-    #result = predict(prompt)
-    return response_text.text
-
-
+    url = f"https://www.googleapis.com/customsearch/v1?key={google_api_key}&cx={cx}&q={prompt}"
     
     response = requests.get(url, headers=headers)
     data = response.json()  # JSON-Daten direkt verarbeiten
@@ -52,7 +42,7 @@ def websearch(prompt):
     items = data.get('items', [])       
     results = [item['snippet'] for item in items]
     result_text = '\n'.join(results)
-    #return results[0]
+    return results[0]
     return result_text
     
     # URL der Google Custom Search API