remove print

app.py

@@ -31,8 +31,16 @@ st.title("API Query App")
 st.write("This app allows you to query the API and retrieve responses.")
 
 user_input = """
-You will be given a text containing some parameters I want you to extract in json format.
-Here are all the paramters: 
+Extrais les paramètres suivants dans un json:
+
+- Date de naissance
+- Prénom 
+- Nom du patient
+
+Dans ta réponse, le json (uniquement) doit apparaitre entre <JSON> et </JSON>.
+Ne répond que par le json entre les balises, si les paramètres n'existent pas, laisse les champs vides.
+
+Voici le texte qui contient les paramètres à extraire:
 
 """
 
@@ -45,12 +53,12 @@ if st.button("Submit"):
         with st.spinner("Extracting text from image..."):
             # Extract text from the uploaded image
             extracted_text = extract_text_from_image(uploaded_image)
-            st.write("Extracted text from image:")
-            st.write(extracted_text)
-        
+            st.write("Extracted text from image.")
+
     with st.spinner("Fetching response from API..."):
         # Query the API with user input
-        input = user_input + " :" 
-        output = query({"inputs": user_input, "parameters": {}})
+        llm_input = user_input + extracted_text + "\n Donne uniquement le json entre balises, pas le texte:"
+        output = query({"inputs": llm_input, "parameters": {}})
         st.success("Response received!")
-        st.write(output[0]["generated_text"])  # Display the response
+        st.write(output)  # Display the response
+