Update app.py

app.py

@@ -35,20 +35,7 @@ def predict(prompt):
     )
 
     chat_session = model.start_chat(
-      history=[
-        {
-          "role": "user",
-          "parts": [
-            "return a json object with the contact details. leave blank if information is not available. here is the json schema:\n\n{\n  \"organization\": \"\",\n  \"address\": \"\",\n  \"phone\": \"\",\n  \"email\": \"\",\n  \"website\": \"\"\n}\n\nyou can find the contact details here: \nImpressum – Aero-Club Bamberg e.V.\nAero-Club Bamberg\nhttps://aeroclub-bamberg.de › impressum\nAero-Club Bamberg e.V.. Zeppelinstraße 18. D-96052 Bamberg. Tel.: 0951 / 45 1 45. Fax: 0951 / 13 22 20. E-Mail: [email protected].\n",
-          ],
-        },
-        {
-          "role": "model",
-          "parts": [
-            "```json\n{\n  \"organization\": \"Aero-Club Bamberg e.V.\",\n  \"address\": \"Zeppelinstraße 18, D-96052 Bamberg\",\n  \"phone\": \"0951 / 45 1 45\",\n  \"email\": \"[email protected]\",\n  \"website\": \"https://aeroclub-bamberg.de\"\n}\n```\n",
-          ],
-        },
-      ]
+      history=[]
     )
       
     response = chat_session.send_message("return a json object with the contact details. leave blank if information is not available. here is the json schema:\n\n{\n  \"organization\": \"\",\n  \"address\": \"\",\n  \"phone\": \"\",\n  \"email\": \"\",\n  \"website\": \"\"\n}\n\nyou can find the contact details here: \n" + prompt)
@@ -93,8 +80,6 @@ def websearch(prompt):
         print("Kein div mit der Klasse 'MjjYud' gefunden.")
         return None
 
-
-
 # Create the Gradio interface
 with gr.Blocks(css=custom_css) as demo:
     with gr.Row():
@@ -106,7 +91,7 @@ with gr.Blocks(css=custom_css) as demo:
         button = gr.Button("Senden")    
 
     # Connect the button to the function
-    button.click(fn=get_impressum_text, inputs=ort_input, outputs=details_output)   
+    button.click(fn=predict, inputs=ort_input, outputs=details_output)   
 
 # Launch the Gradio application
 demo.launch()