Spaces:
Sleeping
Sleeping
Commit
·
d50deb1
1
Parent(s):
408fffa
feat: update descriptions
Browse files- gradio_chat.py +48 -39
gradio_chat.py
CHANGED
|
@@ -17,6 +17,45 @@ model = OpenAIModel(
|
|
| 17 |
api_key=os.environ.get("OPENAI_API_KEY"),
|
| 18 |
)
|
| 19 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
simple_agent = Agent(
|
| 21 |
model,
|
| 22 |
system_prompt="You are a helpful assistant that can answer questions and help with tasks.",
|
|
@@ -160,19 +199,7 @@ async def stream_from_agent(chat_input, chatbot, past_messages, current_image):
|
|
| 160 |
yield gr.Textbox(interactive=True), gr.skip(), past_messages, current_image
|
| 161 |
|
| 162 |
with gr.Blocks() as demo:
|
| 163 |
-
gr.
|
| 164 |
-
"""
|
| 165 |
-
<div style="display: flex; justify-content: center; align-items: center; gap: 2rem; padding: 1rem; width: 100%">
|
| 166 |
-
<img src="https://ai.pydantic.dev/img/logo-white.svg" style="max-width: 200px; height: auto">
|
| 167 |
-
<div>
|
| 168 |
-
<h1 style="margin: 0 0 1rem 0">Image Editing Assistant</h1>
|
| 169 |
-
<h3 style="margin: 0 0 0.5rem 0">
|
| 170 |
-
This assistant edits images according to your instructions.
|
| 171 |
-
</h3>
|
| 172 |
-
</div>
|
| 173 |
-
</div>
|
| 174 |
-
"""
|
| 175 |
-
)
|
| 176 |
|
| 177 |
current_image = gr.State(None)
|
| 178 |
past_messages = gr.State([])
|
|
@@ -181,32 +208,7 @@ with gr.Blocks() as demo:
|
|
| 181 |
label='Image Editing Assistant',
|
| 182 |
type='messages',
|
| 183 |
avatar_images=(None, 'https://ai.pydantic.dev/img/logo-white.svg'),
|
| 184 |
-
examples=
|
| 185 |
-
{
|
| 186 |
-
"text": "Replace the background to the space with stars and planets",
|
| 187 |
-
"files": [
|
| 188 |
-
"https://cdn.prod.website-files.com/66f230993926deadc0ac3a44/66f370d65f158cbbcfbcc532_Crossed%20Arms%20Levi%20Meir%20Clancy.jpg"
|
| 189 |
-
]
|
| 190 |
-
},
|
| 191 |
-
{
|
| 192 |
-
"text": "Change all the balloons to red in the image",
|
| 193 |
-
"files": [
|
| 194 |
-
"https://www.apple.com/tv-pr/articles/2024/10/apple-tv-unveils-severance-season-two-teaser-ahead-of-the-highly-anticipated-return-of-the-emmy-and-peabody-award-winning-phenomenon/images/big-image/big-image-01/1023024_Severance_Season_Two_Official_Trailer_Big_Image_01_big_image_post.jpg.large_2x.jpg"
|
| 195 |
-
]
|
| 196 |
-
},
|
| 197 |
-
{
|
| 198 |
-
"text": "Change coffee to a glass of water",
|
| 199 |
-
"files": [
|
| 200 |
-
"https://previews.123rf.com/images/vadymvdrobot/vadymvdrobot1812/vadymvdrobot181201149/113217373-image-of-smiling-woman-holding-takeaway-coffee-in-paper-cup-and-taking-selfie-while-walking-through.jpg"
|
| 201 |
-
]
|
| 202 |
-
},
|
| 203 |
-
{
|
| 204 |
-
"text": "ENHANCE!",
|
| 205 |
-
"files": [
|
| 206 |
-
"https://m.media-amazon.com/images/M/MV5BNzM3ODc5NzEtNzJkOC00MDM4LWI0MTYtZTkyNmY3ZTBhYzkxXkEyXkFqcGc@._V1_QL75_UX1000_CR0,52,1000,563_.jpg"
|
| 207 |
-
]
|
| 208 |
-
}
|
| 209 |
-
]
|
| 210 |
)
|
| 211 |
|
| 212 |
with gr.Row():
|
|
@@ -233,5 +235,12 @@ with gr.Blocks() as demo:
|
|
| 233 |
outputs=[chat_input, chatbot, past_messages, current_image],
|
| 234 |
)
|
| 235 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 236 |
if __name__ == '__main__':
|
| 237 |
demo.launch()
|
|
|
|
| 17 |
api_key=os.environ.get("OPENAI_API_KEY"),
|
| 18 |
)
|
| 19 |
|
| 20 |
+
INTRO = """
|
| 21 |
+
# Image Editing Assistant
|
| 22 |
+
|
| 23 |
+
### Experience seamless image editing using natural language in a chat-based interface.
|
| 24 |
+
|
| 25 |
+
With this demo, you can:
|
| 26 |
+
- Enhance or upscale an image
|
| 27 |
+
- Remove objects from an image
|
| 28 |
+
- Replace elements within an image
|
| 29 |
+
- Change the background
|
| 30 |
+
"""
|
| 31 |
+
|
| 32 |
+
EXAMPLES = [
|
| 33 |
+
{
|
| 34 |
+
"text": "Replace the background to the space with stars and planets",
|
| 35 |
+
"files": [
|
| 36 |
+
"https://cdn.prod.website-files.com/66f230993926deadc0ac3a44/66f370d65f158cbbcfbcc532_Crossed%20Arms%20Levi%20Meir%20Clancy.jpg"
|
| 37 |
+
]
|
| 38 |
+
},
|
| 39 |
+
{
|
| 40 |
+
"text": "Change all the balloons to red in the image",
|
| 41 |
+
"files": [
|
| 42 |
+
"https://www.apple.com/tv-pr/articles/2024/10/apple-tv-unveils-severance-season-two-teaser-ahead-of-the-highly-anticipated-return-of-the-emmy-and-peabody-award-winning-phenomenon/images/big-image/big-image-01/1023024_Severance_Season_Two_Official_Trailer_Big_Image_01_big_image_post.jpg.large_2x.jpg"
|
| 43 |
+
]
|
| 44 |
+
},
|
| 45 |
+
{
|
| 46 |
+
"text": "Change coffee to a glass of water",
|
| 47 |
+
"files": [
|
| 48 |
+
"https://previews.123rf.com/images/vadymvdrobot/vadymvdrobot1812/vadymvdrobot181201149/113217373-image-of-smiling-woman-holding-takeaway-coffee-in-paper-cup-and-taking-selfie-while-walking-through.jpg"
|
| 49 |
+
]
|
| 50 |
+
},
|
| 51 |
+
{
|
| 52 |
+
"text": "ENHANCE!",
|
| 53 |
+
"files": [
|
| 54 |
+
"https://m.media-amazon.com/images/M/MV5BNzM3ODc5NzEtNzJkOC00MDM4LWI0MTYtZTkyNmY3ZTBhYzkxXkEyXkFqcGc@._V1_QL75_UX1000_CR0,52,1000,563_.jpg"
|
| 55 |
+
]
|
| 56 |
+
}
|
| 57 |
+
]
|
| 58 |
+
|
| 59 |
simple_agent = Agent(
|
| 60 |
model,
|
| 61 |
system_prompt="You are a helpful assistant that can answer questions and help with tasks.",
|
|
|
|
| 199 |
yield gr.Textbox(interactive=True), gr.skip(), past_messages, current_image
|
| 200 |
|
| 201 |
with gr.Blocks() as demo:
|
| 202 |
+
gr.Markdown(INTRO)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 203 |
|
| 204 |
current_image = gr.State(None)
|
| 205 |
past_messages = gr.State([])
|
|
|
|
| 208 |
label='Image Editing Assistant',
|
| 209 |
type='messages',
|
| 210 |
avatar_images=(None, 'https://ai.pydantic.dev/img/logo-white.svg'),
|
| 211 |
+
examples=EXAMPLES
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 212 |
)
|
| 213 |
|
| 214 |
with gr.Row():
|
|
|
|
| 235 |
outputs=[chat_input, chatbot, past_messages, current_image],
|
| 236 |
)
|
| 237 |
|
| 238 |
+
examples = gr.Examples(
|
| 239 |
+
examples=EXAMPLES,
|
| 240 |
+
inputs=[chat_input],
|
| 241 |
+
outputs=[chat_input],
|
| 242 |
+
)
|
| 243 |
+
|
| 244 |
+
|
| 245 |
if __name__ == '__main__':
|
| 246 |
demo.launch()
|