Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import openai
 import fitz  # PyMuPDF for PDF processing
 import base64
+from io import BytesIO
+from PIL import Image
 
 # Variable to store API key
 api_key = ""
@@ -76,7 +78,10 @@ def image_chat(image_file, text_query, temperature, top_p, max_output_tokens):
         ]}
     ]
     response = query_openai(messages, temperature, top_p, max_output_tokens)
-    return response, image_data  # Return image data for display
+    
+    # Convert base64 image to PIL Image for display in Gradio
+    img = Image.open(BytesIO(base64.b64decode(base64_image)))
+    return response, img  # Return image for Gradio to display
 
 # Function to process uploaded PDF input
 def pdf_chat(pdf_file, text_query, temperature, top_p, max_output_tokens):