Update app.py

app.py

@@ -27,7 +27,9 @@ api_key = os.getenv('API_KEY')
 
 # Função principal da API
 def call_fuyu_8b_api(image_path, content, temperature=0.2, top_p=0.7, max_tokens=1024):
-    image_base64 = filepath_to_base64(image_path)
+    if image_path != "No image provided":
+        image_base64 = filepath_to_base64(image_path)
+        content = f"{content} <img src=\"{image_base64}\" />"
     invoke_url = "https://api.nvcf.nvidia.com/v2/nvcf/pexec/functions/9f757064-657f-4c85-abd7-37a7a9b6ee11"
     headers = {
         "Authorization": f"Bearer {api_key}",
@@ -38,7 +40,7 @@ def call_fuyu_8b_api(image_path, content, temperature=0.2, top_p=0.7, max_tokens
     payload = {
         "messages": [
             {
-                "content": f"{content} <img src=\"{image_base64}\" />",
+                "content": content,
                 "role": "user"
             }
         ],
@@ -64,9 +66,10 @@ def call_fuyu_8b_api(image_path, content, temperature=0.2, top_p=0.7, max_tokens
                 decoded_line = line.decode('utf-8')
                 if decoded_line.startswith('data: '):
                     json_str = decoded_line[6:]
-                    json_line = json.loads(json_str)
-                    content_parts = json_line.get("choices", [{}])[0].get("delta", {}).get("content", "")
-                    response_text += content_parts
+                    if json_str:  # Verifica se json_str não está vazio
+                        json_line = json.loads(json_str)
+                        content_parts = json_line.get("choices", [{}])[0].get("delta", {}).get("content", "")
+                        response_text += content_parts
         return response_text
 
 # Interface Gradio
@@ -89,6 +92,7 @@ css = '''
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown("# Fuyu-8B API Explorer with Enhanced Features")
+    content_input = gr.Textbox(lines=2, placeholder="Enter your content here...", label="Content")
     chatbot = gr.Chatbot(label='Chatbot', elem_classes="control-height", height=520)
     task_history = gr.State([])
     with gr.Row():
@@ -97,11 +101,10 @@ with gr.Blocks(css=css) as demo:
         regen_btn = gr.Button("🤔 Regenerate")
         empty_bin = gr.Button("🧹 Clear History")
     with gr.Accordion("Advanced Settings"):
-        content_input = gr.Textbox(lines=2, placeholder="Enter your content here...", label="Content")
         temperature_input = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.2, label="Temperature")
         top_p_input = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.7, label="Top P")
         max_tokens_input = gr.Slider(minimum=1, maximum=1024, step=1, value=1024, label="Max Tokens")
     submit_btn.click(submit_response, [chatbot, task_history, addfile_btn, content_input, temperature_input, top_p_input, max_tokens_input], [chatbot, task_history])
     empty_bin.click(reset_state, [task_history], [chatbot, task_history])
 
-demo.launch()
+demo.launch()