Update app.py

app.py

@@ -243,7 +243,51 @@ def run_process(file, question, model_choice, temperature, top_p, external_api_u
 
     except Exception as e:
         return f"ERROR: {type(e).__name__}: {e}", None
+def run_process_internal_base_v2(file_bytes, filename, mime, question, model_choice, temperature, top_p, batch_size=3):
+    api_key = os.environ.get("GOOGLE_API_KEY", DEFAULT_API_KEY)
+    if not api_key:
+        return "ERROR: Missing GOOGLE_API_KEY.", None
+    genai.configure(api_key=api_key)
+    model_name = INTERNAL_MODEL_MAP.get(model_choice, "gemini-2.5-flash")
+    model = genai.GenerativeModel(model_name=model_name,
+                                  generation_config={"temperature": float(temperature), "top_p": float(top_p)})
+
+    if file_bytes[:4] == b"%PDF":
+        pages = pdf_to_images(file_bytes)
+    else:
+        pages = [Image.open(io.BytesIO(file_bytes))]
+
+    user_prompt = (question or "").strip() or PROMPT_FREIGHT_JSON
+    all_json_results, all_text_results = [], []
+    previous_header_json = None
+
+    def _safe_text(resp):
+        try:
+            return resp.text
+        except:
+            return ""
+
+    for i in range(0, len(pages), batch_size):
+        batch = pages[i:i+batch_size]
+        uploaded = []
+        for im in batch:
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
+                im.save(tmp.name)
+                up = genai.upload_file(path=tmp.name, mime_type="image/png")
+                up = genai.get_file(up.name)
+                uploaded.append(up)
+
+        context_prompt = user_prompt
+        resp = model.generate_content([context_prompt] + uploaded)
+        text = _safe_text(resp)
+        all_text_results.append(text)
+        for up in uploaded:
+            try:
+                genai.delete_file(up.name)
+            except:
+                pass
 
+    return "\n\n".join(all_text_results), None
 # ================== UI ==================
 def main():
     with gr.Blocks(title="OCR Multi-Agent System") as demo: