Update app.py

app.py

@@ -4,7 +4,9 @@ import os
 import base64
 from PIL import Image
 import io
-import json
+
+api_key = os.getenv('API_KEY')
+
 
 def resize_image(image_path, max_size=(800, 800), quality=85):
     with Image.open(image_path) as img:
@@ -18,16 +20,19 @@ def filepath_to_base64(image_path):
     img_base64 = base64.b64encode(img_bytes)
     return img_base64.decode('utf-8')
 
-api_key = os.getenv('API_KEY')
+def format_response(response_body):
+    content = response_body['choices'][0]['message']['content']
+    formatted_content = content.replace("<0x0A>", "\n")
+    return formatted_content
 
 def call_deplot_api(image_path, content, temperature=0.2, top_p=0.7, max_tokens=1024):
     image_base64 = filepath_to_base64(image_path)
     invoke_url = "https://api.nvcf.nvidia.com/v2/nvcf/pexec/functions/3bc390c7-eeec-40f7-a64d-0c6a719985f7"
+    api_key = os.getenv('API_KEY')
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Accept": "application/json",
     }
-
     payload = {
         "messages": [
             {
@@ -40,24 +45,16 @@ def call_deplot_api(image_path, content, temperature=0.2, top_p=0.7, max_tokens=
         "max_tokens": max_tokens,
         "stream": False
     }
-
     session = requests.Session()
     response = session.post(invoke_url, headers=headers, json=payload)
-
     while response.status_code == 202:
         request_id = response.headers.get("NVCF-REQID")
         fetch_url = f"https://api.nvcf.nvidia.com/v2/nvcf/pexec/status/{request_id}"
         response = session.get(fetch_url, headers=headers)
-
     response.raise_for_status()
     response_body = response.json()
-    print(response_body)
-
-    # Processar a resposta conforme necessário
-    # Este exemplo apenas imprime a resposta, mas você pode adaptá-lo conforme necessário
-    return response_body
+    return format_response(response_body)
 
-# Configuração da Interface Gráfica
 content_input = gr.Textbox(lines=2, placeholder="Enter your content here...", label="Content")
 image_input = gr.Image(type="filepath", label="Upload Image")
 temperature_input = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.2, label="Temperature")
@@ -75,7 +72,6 @@ iface = gr.Interface(fn=call_deplot_api,
 <p>
     Utilize Google DePlot to translate images of plots or charts into linearized tables. This one-shot visual language understanding solution offers a unique approach to interpreting visual data.
 </p>
-...
 """
                     )