Update app.py

app.py

@@ -166,8 +166,9 @@ def extract_data_from_pdfs(
             all_text += extract_only_text(reader) + "\n\n"
 
         if include_images == "Include Images":
-            images.extend(extract_images([doc]))
-            img_docs.append(doc.split("/")[-1])
+            imgs = extract_images([doc])
+            images.extend(imgs)
+            img_docs.extend([doc.split("/")[-1] for _ in range(len(imgs))])
 
     progress(
         0.6, "Generating image descriptions and inserting everything into vectorDB"
@@ -269,12 +270,6 @@ def check_validity_and_llm(session_states):
     raise gr.Error("Please extract data first")
 
 
-def get_stats(vectordb):
-    eles = vectordb.get()
-    # words =
-    text_data = [f"Chunks: {len(eles)}", "HIII"]
-    return "\n".join(text_data), "", ""
-
 
 with gr.Blocks(css=CSS, theme=gr.themes.Soft(text_size=sizes.text_md)) as demo:
     vectordb = gr.State()