Update app.py

app.py

@@ -5,7 +5,7 @@ import os
 
 def call_predict_api(image, private_key=None):
     if private_key != os.environ["api_key"]:
-        return "Invalid API key. Please check your key and try again."
+        return "Invalid API key. Please check your key and try again.", "Invalid API key"
     # Convert PIL images to base64 string
     src_buffer = BytesIO()
     image.save(src_buffer, format='PNG')
@@ -20,7 +20,7 @@ def call_predict_api(image, private_key=None):
     if response.status_code != 200:
         raise Exception(f"API Error: {response.text}")
     result = response.json()["result"]
-    return result.replace("\n", "\n\n")
+    return result.replace("\n", "\n\n"), "Success"
 
 if __name__ == "__main__":
     iface = gr.Interface(
@@ -40,7 +40,8 @@ if __name__ == "__main__":
                 lines=1,
             ),
         ],
-        outputs=[gr.Markdown(label="Result")],
+        outputs=[gr.Markdown(label="Result"),
+                 gr.Textbox(label="Status", type="text", visible=False)],
         title="BBC-OCR",
         description="A simple OCR tool to extract text from images.",
     )