Fix model dropdown to show only names and improve output display

app.py

@@ -7,15 +7,43 @@ def get_available_models():
     try:
         response = requests.get("https://text.pollinations.ai/models")
         if response.status_code == 200:
-            return response.json()
+            models_data = response.json()
+            # Extract just the model names if API returns complex structure
+            if isinstance(models_data, list):
+                # If it's a list of strings, return as is
+                if all(isinstance(m, str) for m in models_data):
+                    return models_data
+                # If it's a list of dicts, extract model names/id only
+                elif all(isinstance(m, dict) for m in models_data):
+                    model_names = []
+                    for m in models_data:
+                        # Try to get 'name' or 'id' field, ignore everything else
+                        if 'name' in m and isinstance(m['name'], str):
+                            model_names.append(m['name'])
+                        elif 'id' in m and isinstance(m['id'], str):
+                            model_names.append(m['id'])
+                    return model_names if model_names else [
+                        "openai", "mistral", "mistral-large", 
+                        "claude-3.5-sonnet", "llama-3.3-70b", "gemini"
+                    ]
+            # Fallback to default list
+            return [
+                "openai",
+                "mistral",
+                "mistral-large",
+                "claude-3.5-sonnet",
+                "llama-3.3-70b",
+                "gemini"
+            ]
         else:
-            # Fallback list of models based on API documentation
+            # Fallback list of models
             return [
                 "openai",
                 "mistral",
                 "mistral-large",
                 "claude-3.5-sonnet",
-                "llama-3.3-70b"
+                "llama-3.3-70b",
+                "gemini"
             ]
     except:
         return [
@@ -23,32 +51,50 @@ def get_available_models():
             "mistral",
             "mistral-large",
             "claude-3.5-sonnet",
-            "llama-3.3-70b"
+            "llama-3.3-70b",
+            "gemini"
         ]
 
 # Function to generate text using Pollinations API
-def generate_text(prompt, model, temperature, max_tokens, top_p):
+def generate_text(prompt, model, seed, system, temperature, max_tokens, top_p):
     if not prompt:
         return "Please enter a prompt."
     
     try:
-        # Prepare the API request
+        # Prepare the API request using the same format as user's code
         url = "https://text.pollinations.ai/"
         
         # Build the query parameters
         params = {
             "model": model,
             "prompt": prompt,
-            "temperature": temperature,
-            "max_tokens": int(max_tokens),
-            "top_p": top_p
         }
         
+        # Add optional parameters if provided
+        if seed:
+            params["seed"] = int(seed)
+        if system:
+            params["system"] = system
+        if temperature is not None:
+            params["temperature"] = temperature
+        if max_tokens:
+            params["max_tokens"] = int(max_tokens)
+        if top_p is not None:
+            params["top_p"] = top_p
+        
         # Make the request
         response = requests.get(url, params=params)
         
         if response.status_code == 200:
-            return response.text
+            result_text = response.text
+            
+            # Try to parse as JSON for better formatting
+            try:
+                json_result = json.loads(result_text)
+                return f"```json\n{json.dumps(json_result, indent=2)}\n```"
+            except:
+                # Return as plain text if not JSON
+                return result_text
         else:
             return f"Error: API returned status code {response.status_code}\n{response.text}"
     
@@ -84,6 +130,20 @@ with gr.Blocks(title="Pollinations Text Generator") as demo:
             )
             
             with gr.Accordion("Advanced Settings", open=False):
+                seed_input = gr.Number(
+                    label="Seed (optional)",
+                    value=None,
+                    precision=0,
+                    info="Random seed for reproducible results"
+                )
+                
+                system_input = gr.Textbox(
+                    label="System Prompt (optional)",
+                    placeholder="Enter system instructions...",
+                    lines=2,
+                    info="System-level instructions for the model"
+                )
+                
                 temperature_slider = gr.Slider(
                     minimum=0,
                     maximum=2,
@@ -114,25 +174,54 @@ with gr.Blocks(title="Pollinations Text Generator") as demo:
             generate_btn = gr.Button("Generate", variant="primary")
         
         with gr.Column():
-            output_text = gr.Textbox(
-                label="Generated Text",
-                lines=15,
-                show_copy_button=True
+            output_display = gr.Markdown(
+                value="_Your generated text will appear here..._",
+                label="Generated Text"
             )
+            
+            # Add a readonly textbox for easy copying
+            with gr.Accordion("Copy Output (Plain Text)", open=False):
+                output_copy = gr.Textbox(
+                    label="Copyable Output",
+                    lines=15,
+                    show_copy_button=True,
+                    interactive=False
+                )
     
     gr.Markdown(
         """
         ### About
         This Space uses the [Pollinations API](https://github.com/pollinations/pollinations) for text generation.
         The API supports multiple models and is free to use.
+        
+        **Parameters:**
+        - **Model**: Choose from available AI models
+        - **Seed**: Set a random seed for reproducible outputs
+        - **System**: Provide system-level instructions
+        - **Temperature**: Control response creativity (0=deterministic, 2=very creative)
+        - **Max Tokens**: Set maximum response length
+        - **Top P**: Control diversity via nucleus sampling
         """
     )
     
     # Set up the generate button action
+    def generate_and_display(prompt, model, seed, system, temp, max_tok, top_p):
+        result = generate_text(prompt, model, seed, system, temp, max_tok, top_p)
+        # Return both markdown formatted and plain text versions
+        return result, result
+    
     generate_btn.click(
-        fn=generate_text,
-        inputs=[prompt_input, model_dropdown, temperature_slider, max_tokens_slider, top_p_slider],
-        outputs=output_text
+        fn=generate_and_display,
+        inputs=[
+            prompt_input, 
+            model_dropdown, 
+            seed_input,
+            system_input,
+            temperature_slider, 
+            max_tokens_slider, 
+            top_p_slider
+        ],
+        outputs=[output_display, output_copy]
     )
 
 # Launch the app