JeCabrera commited on
Commit
d8bd84f
·
verified ·
1 Parent(s): 9b78d91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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.UploadButton(
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)