ruff

app_allenai.py

@@ -38,7 +38,6 @@ def safe_chat_fn(message, history, client):
 
 
 with gr.Blocks() as demo:
-
     client = gr.State()
 
     model_dropdown = gr.Dropdown(

app_compare.py

@@ -1,6 +1,4 @@
 import os
-import random
-from typing import Dict, List
 
 import google.generativeai as genai
 import gradio as gr
@@ -31,9 +29,10 @@ def get_all_models():
     ]
 
 
-def generate_discussion_prompt(original_question: str, previous_responses: List[str]) -> str:
+def generate_discussion_prompt(original_question: str, previous_responses: list[str]) -> str:
     """Generate a prompt for models to discuss and build upon previous
-    responses."""
+    responses.
+    """
     prompt = f"""You are participating in a multi-AI discussion about this question: "{original_question}"
 
 Previous responses from other AI models:
@@ -49,7 +48,7 @@ Keep your response focused and concise (max 3-4 paragraphs)."""
     return prompt
 
 
-def generate_consensus_prompt(original_question: str, discussion_history: List[str]) -> str:
+def generate_consensus_prompt(original_question: str, discussion_history: list[str]) -> str:
     """Generate a prompt for final consensus building."""
     return f"""Review this multi-AI discussion about: "{original_question}"
 
@@ -65,7 +64,7 @@ As a final synthesizer, please:
 Keep the final consensus concise but complete."""
 
 
-def chat_with_openai(model: str, messages: List[Dict], api_key: str | None) -> str:
+def chat_with_openai(model: str, messages: list[dict], api_key: str | None) -> str:
     import openai
 
     client = openai.OpenAI(api_key=api_key)
@@ -73,14 +72,14 @@ def chat_with_openai(model: str, messages: List[Dict], api_key: str | None) -> s
     return response.choices[0].message.content
 
 
-def chat_with_anthropic(messages: List[Dict], api_key: str | None) -> str:
+def chat_with_anthropic(messages: list[dict], api_key: str | None) -> str:
     """Chat with Anthropic's Claude model."""
     client = Anthropic(api_key=api_key)
     response = client.messages.create(model="claude-3-sonnet-20240229", messages=messages, max_tokens=1024)
     return response.content[0].text
 
 
-def chat_with_gemini(messages: List[Dict], api_key: str | None) -> str:
+def chat_with_gemini(messages: list[dict], api_key: str | None) -> str:
     """Chat with Gemini Pro model."""
     genai.configure(api_key=api_key)
     model = genai.GenerativeModel("gemini-pro")
@@ -96,7 +95,7 @@ def chat_with_gemini(messages: List[Dict], api_key: str | None) -> str:
 
 
 def chat_with_sambanova(
-    messages: List[Dict], api_key: str | None, model_name: str = "Llama-3.2-90B-Vision-Instruct"
+    messages: list[dict], api_key: str | None, model_name: str = "Llama-3.2-90B-Vision-Instruct"
 ) -> str:
     """Chat with SambaNova's models using their OpenAI-compatible API."""
     client = openai.OpenAI(
@@ -105,13 +104,16 @@ def chat_with_sambanova(
     )
 
     response = client.chat.completions.create(
-        model=model_name, messages=messages, temperature=0.1, top_p=0.1  # Use the specific model name passed in
+        model=model_name,
+        messages=messages,
+        temperature=0.1,
+        top_p=0.1,  # Use the specific model name passed in
     )
     return response.choices[0].message.content
 
 
 def chat_with_hyperbolic(
-    messages: List[Dict], api_key: str | None, model_name: str = "Qwen/Qwen2.5-Coder-32B-Instruct"
+    messages: list[dict], api_key: str | None, model_name: str = "Qwen/Qwen2.5-Coder-32B-Instruct"
 ) -> str:
     """Chat with Hyperbolic's models using their OpenAI-compatible API."""
     client = OpenAI(api_key=api_key, base_url="https://api.hyperbolic.xyz/v1")
@@ -132,7 +134,7 @@ def chat_with_hyperbolic(
 
 
 def multi_model_consensus(
-    question: str, selected_models: List[str], rounds: int = 3, progress: gr.Progress = gr.Progress()
+    question: str, selected_models: list[str], rounds: int = 3, progress: gr.Progress = gr.Progress()
 ) -> list[tuple[str, str]]:
     if not selected_models:
         raise gr.Error("Please select at least one model to chat with.")
@@ -172,7 +174,7 @@ def multi_model_consensus(
 
             chat_history.append((model, response))
         except Exception as e:
-            chat_history.append((model, f"Error: {str(e)}"))
+            chat_history.append((model, f"Error: {e!s}"))
 
     progress(1.0, desc="Done!")
     return chat_history
@@ -180,9 +182,7 @@ def multi_model_consensus(
 
 with gr.Blocks() as demo:
     gr.Markdown("# Model Response Comparison")
-    gr.Markdown(
-        """Select multiple models to compare their responses"""
-    )
+    gr.Markdown("""Select multiple models to compare their responses""")
 
     with gr.Row():
         with gr.Column():

app_crew.py

@@ -1,8 +1,8 @@
-import gradio as gr
 import ai_gradio
+import gradio as gr
 
 demo = gr.load(
-    name='crewai:gpt-4-turbo',
-    crew_type='article',  # or 'support'
+    name="crewai:gpt-4-turbo",
+    crew_type="article",  # or 'support'
     src=ai_gradio.registry,
-)
+)

app_deepseek.py

@@ -1,7 +1,7 @@
-import gradio as gr
 import ai_gradio
+import gradio as gr
 
 demo = gr.load(
-    name='deepseek:deepseek-chat',
+    name="deepseek:deepseek-chat",
     src=ai_gradio.registry,
-)
+)

app_experimental.py

@@ -1,6 +1,5 @@
 import os
 import random
-from typing import Dict, List
 
 import google.generativeai as genai
 import gradio as gr
@@ -31,9 +30,10 @@ def get_all_models():
     ]
 
 
-def generate_discussion_prompt(original_question: str, previous_responses: List[str]) -> str:
+def generate_discussion_prompt(original_question: str, previous_responses: list[str]) -> str:
     """Generate a prompt for models to discuss and build upon previous
-    responses."""
+    responses.
+    """
     prompt = f"""You are participating in a multi-AI discussion about this question: "{original_question}"
 
 Previous responses from other AI models:
@@ -49,7 +49,7 @@ Keep your response focused and concise (max 3-4 paragraphs)."""
     return prompt
 
 
-def generate_consensus_prompt(original_question: str, discussion_history: List[str]) -> str:
+def generate_consensus_prompt(original_question: str, discussion_history: list[str]) -> str:
     """Generate a prompt for final consensus building."""
     return f"""Review this multi-AI discussion about: "{original_question}"
 
@@ -65,7 +65,7 @@ As a final synthesizer, please:
 Keep the final consensus concise but complete."""
 
 
-def chat_with_openai(model: str, messages: List[Dict], api_key: str | None) -> str:
+def chat_with_openai(model: str, messages: list[dict], api_key: str | None) -> str:
     import openai
 
     client = openai.OpenAI(api_key=api_key)
@@ -73,14 +73,14 @@ def chat_with_openai(model: str, messages: List[Dict], api_key: str | None) -> s
     return response.choices[0].message.content
 
 
-def chat_with_anthropic(messages: List[Dict], api_key: str | None) -> str:
+def chat_with_anthropic(messages: list[dict], api_key: str | None) -> str:
     """Chat with Anthropic's Claude model."""
     client = Anthropic(api_key=api_key)
     response = client.messages.create(model="claude-3-sonnet-20240229", messages=messages, max_tokens=1024)
     return response.content[0].text
 
 
-def chat_with_gemini(messages: List[Dict], api_key: str | None) -> str:
+def chat_with_gemini(messages: list[dict], api_key: str | None) -> str:
     """Chat with Gemini Pro model."""
     genai.configure(api_key=api_key)
     model = genai.GenerativeModel("gemini-pro")
@@ -96,7 +96,7 @@ def chat_with_gemini(messages: List[Dict], api_key: str | None) -> str:
 
 
 def chat_with_sambanova(
-    messages: List[Dict], api_key: str | None, model_name: str = "Llama-3.2-90B-Vision-Instruct"
+    messages: list[dict], api_key: str | None, model_name: str = "Llama-3.2-90B-Vision-Instruct"
 ) -> str:
     """Chat with SambaNova's models using their OpenAI-compatible API."""
     client = openai.OpenAI(
@@ -105,13 +105,16 @@ def chat_with_sambanova(
     )
 
     response = client.chat.completions.create(
-        model=model_name, messages=messages, temperature=0.1, top_p=0.1  # Use the specific model name passed in
+        model=model_name,
+        messages=messages,
+        temperature=0.1,
+        top_p=0.1,  # Use the specific model name passed in
     )
     return response.choices[0].message.content
 
 
 def chat_with_hyperbolic(
-    messages: List[Dict], api_key: str | None, model_name: str = "Qwen/Qwen2.5-Coder-32B-Instruct"
+    messages: list[dict], api_key: str | None, model_name: str = "Qwen/Qwen2.5-Coder-32B-Instruct"
 ) -> str:
     """Chat with Hyperbolic's models using their OpenAI-compatible API."""
     client = OpenAI(api_key=api_key, base_url="https://api.hyperbolic.xyz/v1")
@@ -132,7 +135,7 @@ def chat_with_hyperbolic(
 
 
 def multi_model_consensus(
-    question: str, selected_models: List[str], rounds: int = 3, progress: gr.Progress = gr.Progress()
+    question: str, selected_models: list[str], rounds: int = 3, progress: gr.Progress = gr.Progress()
 ) -> list[tuple[str, str]]:
     if not selected_models:
         raise gr.Error("Please select at least one model to chat with.")
@@ -246,7 +249,7 @@ def multi_model_consensus(
                 messages=[{"role": "user", "content": consensus_prompt}], api_key=api_key
             )
     except Exception as e:
-        final_consensus = f"Error getting consensus from {model}: {str(e)}"
+        final_consensus = f"Error getting consensus from {model}: {e!s}"
 
     chat_history.append(("Final Consensus", final_consensus))
 

app_groq.py

@@ -1,20 +1,12 @@
-import os
-
 import ai_gradio
 
 from utils_ai_gradio import get_app
 
 # Get the Groq models from the registry
-GROQ_MODELS_FULL = [
-    k for k in ai_gradio.registry.keys()
-    if k.startswith('groq:')
-]
+GROQ_MODELS_FULL = [k for k in ai_gradio.registry.keys() if k.startswith("groq:")]
 
 # Create display names without the prefix
-GROQ_MODELS_DISPLAY = [
-    k.replace('groq:', '')
-    for k in GROQ_MODELS_FULL
-]
+GROQ_MODELS_DISPLAY = [k.replace("groq:", "") for k in GROQ_MODELS_FULL]
 
 demo = get_app(
     models=GROQ_MODELS_FULL,
@@ -22,7 +14,7 @@ demo = get_app(
     src=ai_gradio.registry,
     dropdown_label="Select Groq Model",
     choices=GROQ_MODELS_DISPLAY,
-    fill_height=True
+    fill_height=True,
 )
 
 if __name__ == "__main__":

app_groq_coder.py

@@ -1,17 +1,12 @@
 import ai_gradio
+
 from utils_ai_gradio import get_app
 
 # Get the Groq models but keep their full names for loading
-GROQ_MODELS_FULL = [
-    k for k in ai_gradio.registry.keys() 
-    if k.startswith('groq:')
-]
+GROQ_MODELS_FULL = [k for k in ai_gradio.registry.keys() if k.startswith("groq:")]
 
 # Create display names without the prefix
-GROQ_MODELS_DISPLAY = [
-    k.replace('groq:', '') 
-    for k in GROQ_MODELS_FULL
-]
+GROQ_MODELS_DISPLAY = [k.replace("groq:", "") for k in GROQ_MODELS_FULL]
 
 # Create and launch the interface using get_app utility
 demo = get_app(
@@ -20,7 +15,7 @@ demo = get_app(
     dropdown_label="Select Groq Model",
     choices=GROQ_MODELS_DISPLAY,  # Display names without prefix
     fill_height=True,
-    coder=True
+    coder=True,
 )
 
 

app_huggingface.py

@@ -44,12 +44,11 @@ def safe_chat_fn(message, history, client):
     try:
         return create_chat_fn(client)(message, history)
     except Exception as e:
-        print(f"Error during chat: {str(e)}")
-        return f"Error during chat: {str(e)}"
+        print(f"Error during chat: {e!s}")
+        return f"Error during chat: {e!s}"
 
 
 with gr.Blocks() as demo:
-
     client = gr.State()
 
     model_dropdown = gr.Dropdown(

app_hyperbolic.py

@@ -1,17 +1,12 @@
 import ai_gradio
+
 from utils_ai_gradio import get_app
 
 # Get the hyperbolic models but keep their full names for loading
-HYPERBOLIC_MODELS_FULL = [
-    k for k in ai_gradio.registry.keys() 
-    if k.startswith('hyperbolic:')
-]
+HYPERBOLIC_MODELS_FULL = [k for k in ai_gradio.registry.keys() if k.startswith("hyperbolic:")]
 
 # Create display names without the prefix
-HYPERBOLIC_MODELS_DISPLAY = [
-    k.replace('hyperbolic:', '') 
-    for k in HYPERBOLIC_MODELS_FULL
-]
+HYPERBOLIC_MODELS_DISPLAY = [k.replace("hyperbolic:", "") for k in HYPERBOLIC_MODELS_FULL]
 
 
 # Create and launch the interface using get_app utility
@@ -20,6 +15,5 @@ demo = get_app(
     default_model=HYPERBOLIC_MODELS_FULL[-1],
     dropdown_label="Select Hyperbolic Model",
     choices=HYPERBOLIC_MODELS_DISPLAY,  # Display names without prefix
-    fill_height=True
+    fill_height=True,
 )
-

app_hyperbolic_coder.py

@@ -1,17 +1,12 @@
 import ai_gradio
+
 from utils_ai_gradio import get_app
 
 # Get the hyperbolic models but keep their full names for loading
-HYPERBOLIC_MODELS_FULL = [
-    k for k in ai_gradio.registry.keys() 
-    if k.startswith('hyperbolic:')
-]
+HYPERBOLIC_MODELS_FULL = [k for k in ai_gradio.registry.keys() if k.startswith("hyperbolic:")]
 
 # Create display names without the prefix
-HYPERBOLIC_MODELS_DISPLAY = [
-    k.replace('hyperbolic:', '') 
-    for k in HYPERBOLIC_MODELS_FULL
-]
+HYPERBOLIC_MODELS_DISPLAY = [k.replace("hyperbolic:", "") for k in HYPERBOLIC_MODELS_FULL]
 
 
 # Create and launch the interface using get_app utility
@@ -21,6 +16,5 @@ demo = get_app(
     dropdown_label="Select Hyperbolic Model",
     choices=HYPERBOLIC_MODELS_DISPLAY,  # Display names without prefix
     fill_height=True,
-    coder=True
+    coder=True,
 )
-

app_openai.py

@@ -1,18 +1,12 @@
 import ai_gradio
-from utils_ai_gradio import get_app
 
+from utils_ai_gradio import get_app
 
 # Get the OpenAI models but keep their full names for loading
-OPENAI_MODELS_FULL = [
-    k for k in ai_gradio.registry.keys() 
-    if k.startswith('openai:')
-]
+OPENAI_MODELS_FULL = [k for k in ai_gradio.registry.keys() if k.startswith("openai:")]
 
 # Create display names without the prefix
-OPENAI_MODELS_DISPLAY = [
-    k.replace('openai:', '') 
-    for k in OPENAI_MODELS_FULL
-]
+OPENAI_MODELS_DISPLAY = [k.replace("openai:", "") for k in OPENAI_MODELS_FULL]
 
 # Create and launch the interface using get_app utility
 demo = get_app(

app_openai_coder.py

@@ -1,18 +1,12 @@
 import ai_gradio
-from utils_ai_gradio import get_app
 
+from utils_ai_gradio import get_app
 
 # Get the OpenAI models but keep their full names for loading
-OPENAI_MODELS_FULL = [
-    k for k in ai_gradio.registry.keys() 
-    if k.startswith('openai:')
-]
+OPENAI_MODELS_FULL = [k for k in ai_gradio.registry.keys() if k.startswith("openai:")]
 
 # Create display names without the prefix
-OPENAI_MODELS_DISPLAY = [
-    k.replace('openai:', '') 
-    for k in OPENAI_MODELS_FULL
-]
+OPENAI_MODELS_DISPLAY = [k.replace("openai:", "") for k in OPENAI_MODELS_FULL]
 
 # Create and launch the interface using get_app utility
 demo = get_app(

app_paligemma.py

@@ -43,8 +43,8 @@ def safe_chat_fn(message, history, client, system_prompt, temperature, max_token
             message, history
         )
     except Exception as e:
-        print(f"Error during chat: {str(e)}")
-        return f"Error during chat: {str(e)}"
+        print(f"Error during chat: {e!s}")
+        return f"Error during chat: {e!s}"
 
 
 with gr.Blocks() as demo:

app_smolagents.py

@@ -1,16 +1,11 @@
 import ai_gradio
+
 from utils_ai_gradio import get_app
 
-SMOLAGENTS_MODELS_FULL = [
-    k for k in ai_gradio.registry.keys() 
-    if k.startswith('smolagents:')
-]
+SMOLAGENTS_MODELS_FULL = [k for k in ai_gradio.registry.keys() if k.startswith("smolagents:")]
 
 
-SMOLAGENTS_MODELS_DISPLAY = [
-    k.replace('smolagents:', '') 
-    for k in SMOLAGENTS_MODELS_FULL
-]
+SMOLAGENTS_MODELS_DISPLAY = [k.replace("smolagents:", "") for k in SMOLAGENTS_MODELS_FULL]
 
 demo = get_app(
     models=SMOLAGENTS_MODELS_FULL,  # Use the full names with prefix

app_transformers.py

@@ -1,9 +1,6 @@
 import gradio as gr
 
-demo = gr.load(
-    name='akhaliq/phi-4',
-    src="spaces"
-)
+demo = gr.load(name="akhaliq/phi-4", src="spaces")
 
 # Disable API access for all functions
 if hasattr(demo, "fns"):
@@ -12,7 +9,3 @@ if hasattr(demo, "fns"):
 
 if __name__ == "__main__":
     demo.launch()
-
-
-
-

pyproject.toml

@@ -59,6 +59,7 @@ line-length = 119
 
 [tool.ruff.lint]
 select = ["E", "F", "I", "N", "S", "T", "UP", "W"]
+ignore = ["T201"]
 
 [tool.ruff.format]
 docstring-code-format = true

utils.py

@@ -1,4 +1,5 @@
-from typing import Callable, Dict, Literal, Union
+from collections.abc import Callable
+from typing import Literal
 
 import gradio as gr
 
@@ -6,7 +7,7 @@ import gradio as gr
 def get_app(
     models: list[str],
     default_model: str,
-    src: Union[Callable[[str, str | None], gr.Blocks], Literal["models"], Dict[str, gr.Blocks]],
+    src: Callable[[str, str | None], gr.Blocks] | Literal["models"] | dict[str, gr.Blocks],
     accept_token: bool = False,
     dropdown_label: str = "Select Model",
     **kwargs,
@@ -37,4 +38,4 @@ def get_app(
     for fn in demo.fns.values():
         fn.api_name = False
 
-    return demo
+    return demo