Update app.py

app.py

@@ -33,13 +33,16 @@ def query_openai(messages, temperature, top_p, max_output_tokens):
 # Function to process image URL input
 def image_url_chat(image_url, text_query, temperature, top_p, max_output_tokens):
     messages = [
-        {"role": "user", "content": [{"type": "input_image", "image_url": image_url}, {"type": "input_text", "text": text_query}]}
+        {"role": "user", "content": [
+            {"type": "image_url", "image_url": {"url": image_url}},  # Corrected format
+            {"type": "text", "text": text_query}
+        ]}
     ]
     return query_openai(messages, temperature, top_p, max_output_tokens)
 
 # Function to process text input
 def text_chat(text_query, temperature, top_p, max_output_tokens):
-    messages = [{"role": "user", "content": text_query}]
+    messages = [{"role": "user", "content": [{"type": "text", "text": text_query}]}]
     return query_openai(messages, temperature, top_p, max_output_tokens)
 
 # Function to process uploaded image input
@@ -50,11 +53,14 @@ def image_chat(image_file, text_query, temperature, top_p, max_output_tokens):
     # Encode image as base64
     with open(image_file, "rb") as img:
         base64_image = base64.b64encode(img.read()).decode("utf-8")
-    
+
     image_data = f"data:image/jpeg;base64,{base64_image}"
-    
+
     messages = [
-        {"role": "user", "content": [{"type": "input_image", "image_url": image_data}, {"type": "input_text", "text": text_query}]}
+        {"role": "user", "content": [
+            {"type": "image_url", "image_url": {"url": image_data}},  # Fixed format
+            {"type": "text", "text": text_query}
+        ]}
     ]
     return query_openai(messages, temperature, top_p, max_output_tokens)
 
@@ -68,7 +74,10 @@ def pdf_chat(pdf_file, text_query, temperature, top_p, max_output_tokens):
     text = "\n".join([page.get_text("text") for page in doc][:5])  # Limit extraction for performance
 
     messages = [
-        {"role": "user", "content": [{"type": "input_text", "text": text}, {"type": "input_text", "text": text_query}]}
+        {"role": "user", "content": [
+            {"type": "text", "text": text},  # Fixed format
+            {"type": "text", "text": text_query}
+        ]}
     ]
     return query_openai(messages, temperature, top_p, max_output_tokens)