Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -63,10 +63,10 @@ def upload(files: Optional[List[str]], chatbot: CHAT_HISTORY) -> CHAT_HISTORY:
|
|
63 |
gemini_files = []
|
64 |
for file in files:
|
65 |
# Verificar si es una imagen o un archivo PDF
|
66 |
-
if file.endswith('.pdf'):
|
67 |
-
gemini_file = upload_to_gemini(file, mime_type="application/pdf")
|
68 |
else:
|
69 |
-
image = Image.open(file).convert('RGB')
|
70 |
image_preview = preprocess_image(image)
|
71 |
if image_preview:
|
72 |
# Display a preview of the uploaded image
|
@@ -104,7 +104,7 @@ def bot(
|
|
104 |
)
|
105 |
|
106 |
text_prompt = [chatbot[-1][0]] if chatbot and chatbot[-1][0] and isinstance(chatbot[-1][0], str) else []
|
107 |
-
image_prompt = [preprocess_image(Image.open(file).convert('RGB')) for file in files if file.endswith(('.jpg', '.jpeg', '.png'))] if files else []
|
108 |
model = genai.GenerativeModel(model_choice)
|
109 |
response = model.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
|
110 |
|
@@ -125,7 +125,7 @@ chatbot_component = gr.Chatbot(
|
|
125 |
text_prompt_component = gr.Textbox(
|
126 |
placeholder="Message...", show_label=False, autofocus=True, scale=8
|
127 |
)
|
128 |
-
upload_button_component = gr.
|
129 |
label="Upload Images or PDF", file_count="multiple", file_types=["image", "pdf"], scale=1
|
130 |
)
|
131 |
run_button_component = gr.Button(value="Run", variant="primary", scale=1)
|
@@ -183,4 +183,4 @@ with gr.Blocks() as demo:
|
|
183 |
queue=False
|
184 |
)
|
185 |
|
186 |
-
demo.queue(max_size=99).launch(debug=False, show_error=True)
|
|
|
63 |
gemini_files = []
|
64 |
for file in files:
|
65 |
# Verificar si es una imagen o un archivo PDF
|
66 |
+
if file.name.endswith('.pdf'):
|
67 |
+
gemini_file = upload_to_gemini(file.name, mime_type="application/pdf")
|
68 |
else:
|
69 |
+
image = Image.open(file.name).convert('RGB')
|
70 |
image_preview = preprocess_image(image)
|
71 |
if image_preview:
|
72 |
# Display a preview of the uploaded image
|
|
|
104 |
)
|
105 |
|
106 |
text_prompt = [chatbot[-1][0]] if chatbot and chatbot[-1][0] and isinstance(chatbot[-1][0], str) else []
|
107 |
+
image_prompt = [preprocess_image(Image.open(file.name).convert('RGB')) for file in files if file.name.endswith(('.jpg', '.jpeg', '.png'))] if files else []
|
108 |
model = genai.GenerativeModel(model_choice)
|
109 |
response = model.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
|
110 |
|
|
|
125 |
text_prompt_component = gr.Textbox(
|
126 |
placeholder="Message...", show_label=False, autofocus=True, scale=8
|
127 |
)
|
128 |
+
upload_button_component = gr.File(
|
129 |
label="Upload Images or PDF", file_count="multiple", file_types=["image", "pdf"], scale=1
|
130 |
)
|
131 |
run_button_component = gr.Button(value="Run", variant="primary", scale=1)
|
|
|
183 |
queue=False
|
184 |
)
|
185 |
|
186 |
+
demo.queue(max_size=99).launch(debug=False, show_error=True)
|