Update app.py

app.py

@@ -11,7 +11,7 @@ client = OpenAI(
     api_key=os.environ.get("NVIDIA_API_KEY"),
 )
 
-MODEL = "openai/gpt-oss-20b"   # GPT-4o via NVIDIA Build
+MODEL = "openai/gpt-4o-mini"   # GPT via NVIDIA Build — change this to match your NVIDIA model name
 
 # ── System prompts ────────────────────────────────────────────────────────────
 SYSTEM_PROMPTS = {
@@ -81,8 +81,10 @@ def pdf_to_base64_images(pdf_path: str) -> list:
 def build_messages(system, history, user_text, image_b64_list=None):
     messages = [{"role": "system", "content": system}]
     for human, assistant in history:
-        messages.append({"role": "user",      "content": human})
-        messages.append({"role": "assistant", "content": assistant})
+        if human is not None:
+            messages.append({"role": "user",      "content": str(human)})
+        if assistant is not None:
+            messages.append({"role": "assistant", "content": str(assistant)})
 
     if image_b64_list:
         content = []
@@ -119,22 +121,27 @@ def chat(message, history, mode, file_upload, age, gender, weight, conditions, a
     display_msg  = message.strip() or "📄 File uploaded — please analyze"
 
     try:
-        image_b64_list = None
-
+        # NVIDIA Build does not support vision — text only
         if file_upload is not None:
             ext = os.path.splitext(file_upload)[1].lower()
             if ext == ".pdf":
-                image_b64_list = pdf_to_base64_images(file_upload)[:6]
-                display_msg = f"📄 PDF uploaded" + (f" — {message.strip()}" if message.strip() else "")
+                # Extract text from PDF instead
+                doc = fitz.open(file_upload)
+                pdf_text = ""
+                for page in doc:
+                    pdf_text += page.get_text()
+                doc.close()
+                user_text = f"The patient has uploaded a lab report. Here is the extracted text:\n\n{pdf_text[:4000]}\n\n{user_text}"
+                display_msg = "📄 PDF uploaded" + (f" — {message.strip()}" if message.strip() else "")
             elif ext in [".jpg", ".jpeg", ".png", ".webp", ".bmp"]:
-                image_b64_list = [encode_image_to_base64(file_upload)]
-                display_msg = f"🖼️ Image uploaded" + (f" — {message.strip()}" if message.strip() else "")
+                user_text = "The patient has uploaded an image of a medical document (lab report/prescription/X-ray). Unfortunately I cannot read images directly — please type out the key values from the report and I will analyze them for you.\n\n" + user_text
+                display_msg = "🖼️ Image uploaded" + (f" — {message.strip()}" if message.strip() else "")
             else:
-                history.append((display_msg, "❌ Unsupported file type. Please upload PDF, JPG, or PNG."))
+                history.append((message or "File upload", "❌ Unsupported file type. Please upload PDF, JPG, or PNG."))
                 yield "", history
                 return
 
-        messages = build_messages(system, history, user_text, image_b64_list)
+        messages = build_messages(system, history, user_text)
 
         # ── Streaming response ──
         stream = client.chat.completions.create(
@@ -283,4 +290,4 @@ with gr.Blocks(css=CSS, title="Sehat Guard — AI Medical Assistant") as demo:
     send_btn.click(fn=chat, inputs=inputs, outputs=outputs)
     user_input.submit(fn=chat, inputs=inputs, outputs=outputs)
 
-demo.launch()
+demo.launch()