add playai

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 
+from app_playai import demo as demo_playai
 from app_allenai import demo as demo_allenai
 from app_claude import demo as demo_claude
 from app_experimental import demo as demo_experimental
@@ -20,6 +21,8 @@ from app_together import demo as demo_together
 from app_xai import demo as demo_grok
 
 with gr.Blocks(fill_height=True) as demo:
+    with gr.Tab("PlayAI"):
+        demo_playai.render()
     with gr.Tab("Grok"):
         demo_grok.render()
     with gr.Tab("Hyperbolic"):

app_huggingface.py

@@ -1,14 +1,16 @@
 from gradio_client import Client
 import gradio as gr
+import os
+
 
 MODELS = {
     "SmolVLM-Instruct": "akhaliq/SmolVLM-Instruct"
 }
 
 def create_chat_fn(client):
-    def chat(message, history):
+    def chat(message, history, files=[]):
         response = client.predict(
-            message={"text": message, "files": []},
+            message={"text": message, "files": files},
             system_prompt="You are a helpful AI assistant.",
             temperature=0.7,
             max_new_tokens=1024,
@@ -22,8 +24,8 @@ def create_chat_fn(client):
 
 def set_client_for_session(model_name, request: gr.Request):
     headers = {}
-    if request and hasattr(request, 'request') and hasattr(request.request, 'headers'):
-        x_ip_token = request.request.headers.get('x-ip-token')
+    if request and hasattr(request, 'headers'):
+        x_ip_token = request.headers.get('x-ip-token')
         if x_ip_token:
             headers["X-IP-Token"] = x_ip_token
     
@@ -32,7 +34,11 @@ def set_client_for_session(model_name, request: gr.Request):
 def safe_chat_fn(message, history, client):
     if client is None:
         return "Error: Client not initialized. Please refresh the page."
-    return create_chat_fn(client)(message, history)
+    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)}"
 
 with gr.Blocks() as demo:
     
@@ -52,22 +58,18 @@ with gr.Blocks() as demo:
     )
     
     # Update client when model changes
-    def update_model(model_name, request):
-        return set_client_for_session(model_name, request)
-    
     model_dropdown.change(
-        fn=update_model,
+        fn=set_client_for_session,
         inputs=[model_dropdown],
-        outputs=[client],
+        outputs=[client]
     )
     
     # Initialize client on page load
     demo.load(
         fn=set_client_for_session,
-        inputs=gr.State("SmolVLM-Instruct"),
-        outputs=client,
+        inputs=[gr.State("SmolVLM-Instruct")],
+        outputs=[client]
     )
 
 demo = demo
 
-demo.launch()

app_playai.py

@@ -0,0 +1,7 @@
+import gradio as gr
+import playai_gradio
+
+demo =gr.load(
+    name='PlayDialog',
+    src=playai_gradio.registry,
+)

requirements.txt

@@ -396,3 +396,4 @@ websockets==12.0
     # via gradio-client
 xai-gradio==0.0.2
     # via anychat (pyproject.toml)
+playai-gradio @ git+https://github.com/AK391/playai-gradio.git