Update app.py

app.py

@@ -5,7 +5,15 @@ from urllib.parse import urljoin
 
 def parse_links(prompt):
 #anfrage ='fill in the missing contact information. do not reference the json object. do not use html tags inside the json object. it must be valid json. do not return explanaitons or any other text. thanks for your efforts:)'
-
+anfrage ="""
+fill in the missing contact information. do not reference the json object. do not use html tags inside the json object. it must be valid json. do not return explanaitons or any other text. thanks for your efforts:)
+{
+  "Name": f"{prompt}",
+  "Email": "",
+  "Website": "",
+  "Phone": ""
+}
+"""
     try:
         url = "http://specialist-it.de:3001/api/search"
 
@@ -20,7 +28,7 @@ def parse_links(prompt):
             },
             "optimizationMode": "balanced",
             "focusMode": "webSearch",
-            "query": f"{prompt}. outputformat=json",
+            "query": f"{anfrage}. outputformat=json",
             "history": [
                 ["human", "Hi, how are you?"],
                 ["assistant", "I am doing well, how can I help you today?"]
@@ -38,8 +46,19 @@ def parse_links(prompt):
         #data = response.json()
         # Extract the message
         #ergebnis = data.get('message')
-        #return ergebnis
-        return response.json().get('message')
+        ergebnis = response.json().get('message')
+
+        # Split the response to get the JSON part only 
+        json_part = ergebnis.split('}\n')[0] + '}' 
+        # Load the JSON part to validate it's correctly formatted 
+        try: 
+            data = json.loads(json_part) 
+            print(json.dumps(data, indent=2)) 
+            return json.dumps(data, indent=2)
+        except json.JSONDecodeError as e: 
+            print(f"Invalid JSON: {e}"
+        
+        #return response.json().get('message')
         # Rückgabe als Dictionary, um es mit gr.JSON kompatibel zu machen
         #return {"body_text": body_text}