Update app.py

app.py

@@ -5,6 +5,7 @@ import os
 from PIL import Image
 import io
 from langchain_openai import ChatOpenAI
+from langchain_community.callbacks import get_openai_callback
 import replicate
 
 # Set up environment variables for API keys
@@ -44,20 +45,20 @@ def generate_prompt_gpt_3_turbo(image,instructions):
     
     # Format the prompt with the image description
     prompt = instructions.format(image_description=image_description)
-    
-    # Invoke LLM model and get response
-    response = llm.invoke(prompt).content
+    with get_openai_callback() as cb:
+        # Invoke LLM model and get response
+        response = llm.invoke(prompt).content
     
     # Convert response to JSON if necessary
     response_dict = json.loads(response)
     
-    return response_dict, image_description
+    return response_dict, image_description, cb
 
 # Define the Gradio interface
 iface = gr.Interface(
     fn=generate_prompt_gpt_3_turbo,
     inputs=[gr.Image(type="pil"),gr.Textbox(value = instructions)],
-    outputs=[gr.JSON(),gr.Textbox()],
+    outputs=[gr.JSON(),gr.Textbox(),gr.Textbox()],
     title="Image to Text Generator",
     description="Upload an image to generate descriptive text based on the image."
 )