Update raw.py

raw.py

@@ -184,8 +184,28 @@ with gr.Blocks(title="FLUX Turbo Upscaler", fill_height=True) as iface:
             
     
     with gr.Row():
-        prompt_box = gr.Textbox(lines=4, value="Write a straightforward caption for this image. Begin with the main subject and medium. Mention pivotal elements—people, objects, scenery—using confident, definite language. Focus on concrete details like color, shape, texture, and spatial relationships. Show how elements interact. Omit mood and speculative wording. If text is present, quote it exactly. Note any watermarks, signatures, or compression artifacts. Never mention what's absent, resolution, or unobservable details. Vary your sentence structure and keep the description concise, without starting with “This image is…” or similar phrasing.", visible=False)
-
+        with gr.Accordion("Generation settings", open=False):
+            prompt_box = gr.Textbox(lines=4, value="Write a straightforward caption for this image. Begin with the main subject and medium. Mention pivotal elements—people, objects, scenery—using confident, definite language. Focus on concrete details like color, shape, texture, and spatial relationships. Show how elements interact. Omit mood and speculative wording. If text is present, quote it exactly. Note any watermarks, signatures, or compression artifacts. Never mention what's absent, resolution, or unobservable details. Vary your sentence structure and keep the description concise, without starting with “This image is…” or similar phrasing.", visible=False)
+            temperature_slider = gr.Slider(
+                minimum=0.0, maximum=2.0, value=0.6, step=0.05,
+                label="Temperature",
+                info="Higher values make the output more random, lower values make it more deterministic.",
+                visible=False
+            )
+            top_p_slider = gr.Slider(
+                minimum=0.0, maximum=1.0, value=0.9, step=0.01,
+                label="Top-p",
+                visible=False
+            )
+            max_tokens_slider = gr.Slider(
+                minimum=1, maximum=2048, value=512, step=1,
+                label="Max New Tokens",
+                info="Maximum number of tokens to generate.  The model will stop generating if it reaches this limit.",
+                visible=False
+            )
+        
+        log_prompt = gr.Checkbox(value=True, label="Help improve JoyCaption by logging your text query", visible=False)
+        
         gr.Markdown("**Tips:** 8 steps is all you need!")
     
     generate_button.click(
@@ -194,8 +214,8 @@ with gr.Blocks(title="FLUX Turbo Upscaler", fill_height=True) as iface:
         outputs=[generated_image]
     )
     caption_button.click(
-		caption,
-		inputs=[control_image, prompt_box, 0.6, 0.9, 512, True],
+		fn=caption,
+		inputs=[control_image, prompt_box, temperature_slider, top_p_slider, max_tokens_slider, log_prompt],
 		outputs=output_caption,
 	)
 # Launch the app