Update app.py

app.py

@@ -9,11 +9,16 @@ import json
 import groq
 import os
 
+google_api_key = os.getenv('google_search')
+API_URL = "https://blavken-flowiseblav.hf.space/api/v1/prediction/fbc118dc-ec00-4b59-acff-600648958be3"
+api_key = os.getenv('groq')
+client = groq.Client(api_key=api_key)
+
 custom_css = """
 #md {
-    height: 400px;  
+    height: 200px;  
     font-size: 30px;
-    background: #202020;
+    background: #121212;
     padding: 20px;
     color: white;
     border: 1 px solid white;
@@ -21,18 +26,24 @@ custom_css = """
 }
 """
 
-api_key = os.getenv('groq')
-client = groq.Client(api_key=api_key)
+def query(payload):
+    response = requests.post(API_URL, json=payload)
+    return response.json()
 
-def qwen(jsondata):
-    client = Client("Qwen/Qwen2.5-72B-Instruct")
-    result = client.predict(
-		query= f"return valid json \n {jsondata}",
-		history=[],
-		system="You are Qwen, created by Alibaba Cloud. You are a helpful assistant.",
-		api_name="/model_chat"
-    )
-    return result
+def google_search(payloads):
+    output = query({
+        "question": f"{payloads}",
+    })
+    print(output)
+    #return result_text
+      
+    # Formuliere die Antwort
+    #search_query = f"{payloads} antworte kurz und knapp. antworte auf deutsch. du findest die antwort hier:\n {output}"
+    #result = predict(search_query)
+    texte=""
+    for o in output:
+        texte +=o
+    return texte
 
 def llm(message):
     message = f"return a json object with the keys: name,email,phone,website \n the values can be found here, leave blank if value is not available:\n {message} \n return a json object only. no text, no explanaition"
@@ -48,6 +59,16 @@ def llm(message):
     except Exception as e:
         return f"Error in response generation: {str(e)}"
 
+def qwen(jsondata):
+    client = Client("Qwen/Qwen2.5-72B-Instruct")
+    result = client.predict(
+		query= f"return valid json \n {jsondata}",
+		history=[],
+		system="You are Qwen, created by Alibaba Cloud. You are a helpful assistant.",
+		api_name="/model_chat"
+    )
+    return result
+
 def list_of_clubs(ort):
     base_url = "https://vereine-in-deutschland.net"
     all_links_text = []
@@ -89,7 +110,12 @@ def process_ort(ort):
     links_text = list_of_clubs(ort)
     return links_text
     vereine = []
-    
+    for verein in links_text:
+        prompt=f"impressum {verein}",
+        result = google_search(prompt)
+        json_data = llm(result)
+        vereine.append(result)
+        
     for verein in links_text:
         client = Client("mgokg/gemini-2.0-flash-exp")
         result = client.predict(
@@ -127,7 +153,7 @@ with gr.Blocks(css=custom_css) as demo:
         button = gr.Button("Senden")    
 
     # Connect the button to the function
-    button.click(fn=process_ort, inputs=ort_input, outputs=details_output)   
+    button.click(fn=google_search, inputs=ort_input, outputs=details_output)   
 
 # Launch the Gradio application
 demo.launch()