Update app.py

app.py

@@ -215,24 +215,28 @@ DEMO_LIST = [
     }
 ]
 
-# HF Inference Client
 HF_TOKEN = os.getenv('HF_TOKEN')
 GROQ_API_KEY = os.getenv('GROQ_API_KEY')
+FIREWORKS_API_KEY = os.getenv('FIREWORKS_API_KEY')
 
 def get_inference_client(model_id):
-    """Return an InferenceClient configured for Hugging Face or Groq."""
+    """Return an InferenceClient configured for Hugging Face, Groq, or Fireworks AI."""
     if model_id == "moonshotai/Kimi-K2-Instruct":
         return InferenceClient(
             base_url="https://api.groq.com/openai/v1",
             api_key=GROQ_API_KEY
         )
+    elif model_id.startswith("fireworks/"):
+        return InferenceClient(
+            base_url="https://api.fireworks.ai/inference/v1",
+            api_key=FIREWORKS_API_KEY
+        )
     else:
         return InferenceClient(
             model=model_id,
             api_key=HF_TOKEN
         )
 
-
 # Type definitions
 History = List[Tuple[str, str]]
 Messages = List[Dict[str, str]]