Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import google.genai as genai
+from google.genai import types
 from PIL import Image
 import os
 import textract
@@ -117,15 +118,19 @@ def single_submit_func(prompt, files, model, temperature, top_p, max_tokens, api
         inputs = [prompt] + processed_inputs
         warning = ""
     
-    # Generation configuration
-    generation_config = {
-        "temperature": temperature,
-        "top_p": top_p,
-        "max_output_tokens": max_tokens,
-    }
-    
+    contents = [
+        types.Content(
+            role="user",
+            parts=[
+                types.Part.from_text(text=prompt),
+            ],
+        ),
+    ]
+    generate_content_config = types.GenerateContentConfig(
+        response_mime_type="text/plain",
+    )    
     try:
-        response = client.models.generate_content(inputs, model=model, config=generation_config)
+        response = client.models.generate_content(model=model, contents=contents, config=generation_config)
         response_text = warning
         response_images = []