Update app.py

app.py

@@ -7,23 +7,15 @@ import textract
 
 # List of available models (including experimental and recent ones)
 models = [
-    "gemini-2.5-flash-preview-04-17",
-    "gemini-2.5-pro-preview-03-25",
-    "gemini-2.0-flash",
-    "gemini-2.0-flash-lite",
-    "gemini-2.0-flash-thinking-exp-01-21",
-    "gemini-1.5-pro",
+    "gemini-2.5-flash",
+    "gemini-2.5-pro",
     "gemini-2.0-flash-exp-image-generation"
 ]
 
 # Model types for handling inputs
 model_types = {
-    "gemini-2.5-flash-preview-04-17": "text",
-    "gemini-2.5-pro-preview-03-25": "text",
-    "gemini-2.0-flash": "text",
-    "gemini-2.0-flash-lite": "text",
-    "gemini-2.0-flash-thinking-exp-01-21": "text",
-    "gemini-1.5-pro": "text",
+    "gemini-2.5-flash": "text",
+    "gemini-2.5-pro": "text",
     "gemini-2.0-flash-exp-image-generationn": "multimodal"
 }
 
@@ -52,8 +44,7 @@ def process_files(files, model_type):
     return inputs
 
 # Chat submit function
-def chat_submit_func(message, files, chat_history, model, temperature, top_p, max_tokens, api_key):
-    print(model)
+def chat_submit_func(message, files, chat_history, model, temperature, top_p, max_tokens, api_key, response_type="text/plain"):
     client = genai.Client(api_key=api_key)
     
     # Prepare inputs
@@ -71,6 +62,8 @@ def chat_submit_func(message, files, chat_history, model, temperature, top_p, ma
         "top_p": top_p,
         "max_output_tokens": max_tokens,
     }
+    if response_type:
+        generation_config["response_mime_type"] = response_type
     
     try:
         response = client.models.generate_content(inputs, model=model, config=generation_config)
@@ -104,8 +97,7 @@ def chat_submit_func(message, files, chat_history, model, temperature, top_p, ma
         return chat_history, ""
 
 # Single response submit function
-def single_submit_func(prompt, files, model, temperature, top_p, max_tokens, api_key):
-    print(model)
+def single_submit_func(prompt, files, model, temperature, top_p, max_tokens, api_key, response_type="text/plain"):
     client = genai.Client(api_key=api_key)
     
     # Prepare inputs
@@ -127,7 +119,7 @@ def single_submit_func(prompt, files, model, temperature, top_p, max_tokens, api
         ),
     ]
     generate_content_config = types.GenerateContentConfig(
-        response_mime_type="text/plain",
+        response_mime_type=response_type,
     )    
     try:
         response = client.models.generate_content(model=model, contents=contents, config=generate_content_config)
@@ -160,7 +152,8 @@ with gr.Blocks(title="Gemini API Interface") as app:
         model_selector = gr.Dropdown(choices=models, label="Select Model", value=models[0])
         temperature = gr.Slider(0, 1, value=0.7, label="Temperature", step=0.01)
         top_p = gr.Slider(0, 1, value=0.9, label="Top P", step=0.01)
-        max_tokens = gr.Number(value=512, label="Max Tokens", minimum=1)
+        max_tokens = gr.Number(value=65536, label="Max Tokens", minimum=1)
+        response_type_selector = gr.Radio(choices=["text/plain", "application/json"], value="text/plain", label="Response MIME Type")
     
     # Tabs for Chat and Single Response (hidden until key is validated)
     with gr.Tabs(visible=False) as tabs:
@@ -194,15 +187,15 @@ with gr.Blocks(title="Gemini API Interface") as app:
     # Chat submission
     chat_submit_btn.click(
         chat_submit_func,
-        inputs=[chat_input, chat_files, chat_display, model_selector, temperature, top_p, max_tokens, api_key_input],
+        inputs=[chat_input, chat_files, chat_display, model_selector, temperature, top_p, max_tokens, api_key_input,response_type_selector],
         outputs=[chat_display, chat_status]
     )
     
     # Single response submission
     single_submit_btn.click(
         single_submit_func,
-        inputs=[single_input, single_files, model_selector, temperature, top_p, max_tokens, api_key_input],
+        inputs=[single_input, single_files, model_selector, temperature, top_p, max_tokens, api_key_input, response_type_selector],
         outputs=[single_text_output, single_image_output]
     )
 
-app.launch()
+app.launch()