Update app.py

app.py

@@ -9,6 +9,7 @@ from audiorecorder import audiorecorder
 # Umgebungsvariablen laden
 groq_api_key = os.getenv('groqwhisper')
 gemini_api_key = os.getenv('geminiapi')
+google_api_key = os.getenv('google_search')
 
 if not groq_api_key or not gemini_api_key:
     st.error("Bitte setze die Umgebungsvariablen 'groqwhisper' und 'geminiapi'")
@@ -41,6 +42,24 @@ if "chat_session" not in st.session_state:
 if "display_history" not in st.session_state:
     st.session_state.display_history = []
 
+
+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}"   
+    response = requests.get(url, headers=headers)
+    data = response.json()  # JSON-Daten direkt verarbeiten
+    # Extrahieren des Textes aus den Ergebnissen
+    items = data.get('items', []) 
+    results = [item['snippet'] for item in items]
+    result_text = '\n'.join(results)   
+    # Formuliere die Antwort
+    search_query = f"{prompt} antworte kurz und knapp. antworte auf deutsch. du findest die antwort hier: {result_text}"
+    result = predict(search_query)
+    return result
+
+
 def process_audio(audio_data):
     """Verarbeitet Audiodaten und gibt Transkript zurück."""
     try:
@@ -90,8 +109,9 @@ if audio_bytes:
                 if transcription.startswith("Fehler:"):
                     st.error(transcription)
                 else:
-                    #st.session_state.display_history.append(("user", transcription))
-                    full_prompt = f"{transcription}\nAntworte immer auf Deutsch"
+                    #st.session_state.display_history.append(("user", transcription))                    
+                    websearch = websearch(transcription)
+                    full_prompt = f"{transcription}\nAntworte immer auf Deutsch\n Du findest die Antwort hier:\n{webearch}"
                     response = st.session_state.chat_session.send_message(full_prompt)
                     response_text = response.candidates[0].content.parts[0].text if response.candidates else "Keine Antwort"
                     messages.chat_message("assistant").write(f"{response_text}")
@@ -104,6 +124,8 @@ if audio_bytes:
 # Texteingabe verarbeiten
 user_input = st.text_input("Schreibe deine Frage:", key="user_input")
 if user_input:
+    
+    
     #st.session_state.display_history.append(("user", user_input))
     full_prompt = f"{user_input}\nAntworte immer auf Deutsch\n antworte kurz und knapp."
     response = st.session_state.chat_session.send_message(full_prompt)