Create app.py

app.py

@@ -0,0 +1,81 @@
+import itertools
+import gradio as gr
+import requests
+import os
+from gradio.themes.utils import sizes
+
+
+def respond(message, history):
+
+    if len(message.strip()) == 0:
+        return "ERROR the question should not be empty"
+
+    local_token = os.getenv('API_TOKEN')
+    local_endpoint = os.getenv('API_ENDPOINT')
+
+    if local_token is None or local_endpoint is None:
+        return "ERROR missing env variables"
+
+    # Add your API token to the headers
+    headers = {
+        'Content-Type': 'application/json',
+        'Authorization': f'Bearer {local_token}'
+    }
+
+    #prompt = list(itertools.chain.from_iterable(history))
+    #prompt.append(message)
+    #q = {"inputs": [prompt]}
+    q = {
+          "dataframe_split": {
+            "columns": [
+              "prompt",
+              "num_inference_steps"
+            ],
+            "data": [
+              [
+                  {message},25
+              ]
+                    ]
+          }
+        }
+    
+    try:
+        response = requests.post(
+            local_endpoint, json=q, headers=headers, timeout=100)
+        response_data = response.json()
+        #print(response_data)
+        response_data=response_data["predictions"][0]
+        #print(response_data)
+
+    except Exception as error:
+        response_data = f"ERROR status_code: {type(error).__name__}"
+        # + str(response.status_code) + " response:" + response.text
+
+    # print(response.json())
+    return response_data
+
+
+theme = gr.themes.Soft(
+    text_size=sizes.text_sm,radius_size=sizes.radius_sm, spacing_size=sizes.spacing_sm,
+)
+
+
+demo = gr.ChatInterface(
+    respond,
+    chatbot=gr.Chatbot(show_label=False, container=False, show_copy_button=True, bubble_full_width=True),
+    textbox=gr.Textbox(placeholder="生成する画像を指示",
+                       container=False, scale=7),
+    title="Databricks Personalized Image demo - Generate personalized image using model serving endpoint",
+    description="[Databricksにおける生成AIを用いたブランドに沿う画像の生成](https://qiita.com/taka_yayoi/items/8d3473847d9ccc8ca00c)",
+    examples=[["A photo of an orange bcnchr chair"],
+              ["A photo of an blue hsmnchr chair"],
+              ["A photo of an red rckchr chair"],],
+    cache_examples=False,
+    theme=theme,
+    retry_btn=None,
+    undo_btn=None,
+    clear_btn="Clear",
+)
+
+if __name__ == "__main__":
+    demo.launch()