Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ def process_text(text_input, unit):
|
|
| 32 |
return completion.choices[0].message.content
|
| 33 |
return ""
|
| 34 |
|
| 35 |
-
def encode_image_to_base64(
|
| 36 |
buffered = io.BytesIO()
|
| 37 |
image.save(buffered, format="JPEG")
|
| 38 |
img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
|
|
@@ -42,7 +42,7 @@ def process_image(image_input, unit):
|
|
| 42 |
if image_input is not None:
|
| 43 |
#with open(image_input.name, "rb") as f:
|
| 44 |
# base64_image = base64.b64encode(f.read()).decode("utf-8")
|
| 45 |
-
base64_image = encode_image_to_base64(
|
| 46 |
response = client.chat.completions.create(
|
| 47 |
model=MODEL,
|
| 48 |
messages=[
|
|
@@ -80,13 +80,13 @@ with gr.Blocks() as iface:
|
|
| 80 |
with gr.Row():
|
| 81 |
output_box = gr.Markdown(label="分析") # Create an output textbox
|
| 82 |
with gr.Row():
|
| 83 |
-
|
| 84 |
text_input = gr.Textbox(label="输入") # Create a text input box
|
| 85 |
with gr.Row():
|
| 86 |
submit_btn = gr.Button("🚀 确认") # Create a submit button
|
| 87 |
clear_btn = gr.ClearButton(output_box, value="🗑️ 清空") # Create a clear button
|
| 88 |
|
| 89 |
# Set up the event listeners
|
| 90 |
-
submit_btn.click(main, inputs=[text_input,
|
| 91 |
|
| 92 |
iface.launch() # Launch the Gradio interface
|
|
|
|
| 32 |
return completion.choices[0].message.content
|
| 33 |
return ""
|
| 34 |
|
| 35 |
+
def encode_image_to_base64(image_input):
|
| 36 |
buffered = io.BytesIO()
|
| 37 |
image.save(buffered, format="JPEG")
|
| 38 |
img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
|
|
|
|
| 42 |
if image_input is not None:
|
| 43 |
#with open(image_input.name, "rb") as f:
|
| 44 |
# base64_image = base64.b64encode(f.read()).decode("utf-8")
|
| 45 |
+
base64_image = encode_image_to_base64(image_input)
|
| 46 |
response = client.chat.completions.create(
|
| 47 |
model=MODEL,
|
| 48 |
messages=[
|
|
|
|
| 80 |
with gr.Row():
|
| 81 |
output_box = gr.Markdown(label="分析") # Create an output textbox
|
| 82 |
with gr.Row():
|
| 83 |
+
image_input = gr.Image(type="filepath", label="上传图片") # Create an image upload button
|
| 84 |
text_input = gr.Textbox(label="输入") # Create a text input box
|
| 85 |
with gr.Row():
|
| 86 |
submit_btn = gr.Button("🚀 确认") # Create a submit button
|
| 87 |
clear_btn = gr.ClearButton(output_box, value="🗑️ 清空") # Create a clear button
|
| 88 |
|
| 89 |
# Set up the event listeners
|
| 90 |
+
submit_btn.click(main, inputs=[text_input, image_input, unit], outputs=output_box)
|
| 91 |
|
| 92 |
iface.launch() # Launch the Gradio interface
|