Update app.py

app.py

@@ -40,112 +40,116 @@ class PDFTextExtractor:
                         Extract the Patient Admission Form (KMHIPF002V3) from this image and format it in clean markdown.
                         Follow this exact markdown structure:
 
-# PATIENT ADMISSION FORM
-## DR.KAMAKSHI MEMORIAL HOSPITAL, PALLIKARANAI, CHENNAI.
-
-### PATIENT PROFILE
-*Please paste the sticker within the box*
-
-* UHID: ______
-* Patient Name: ______
-* Age/Gender: ______
-* Doctor Name: ______
-
-### BASIC INFORMATION
-* Date & Time of Admission: [DD/MM/YYYY]
-* Date of Birth: [DD/MM/YYYY]
-
-### IDENTIFICATION
-**ID Proof Already Registered**: □ Yes □ No
-
-**Type of ID**:
-* □ Aadhar
-* □ Passport
-* □ Voter ID
-* □ Driving License
-* □ Others
-
-ID No.: ________________
-Contact No.: ________________
-
-### MEDICAL DETAILS
-* Provisional Diagnosis:
-* Reason for Admission:
-* Plan of Care:
-* Expected Outcome*:
-
-### CONSULTANT DETAILS
-* Primary Consultant Name:
-* Speciality:
-
-### PATIENT REFERENCE INFORMATION
-*(To be filled by Front Office)*
-
-**Reference Via**:
-* □ Doctor
-* □ Hospital
-* □ Ambulance
-* □ DRKMH Employee
-* □ Self / Walk In
-
-**Referrer Details**:
-* Name: ________________
-* Contact No.: ________________
-
-### TYPE OF ADMISSION
-* □ Emergency
-* □ Elective
-* □ MLC
-* □ Surgery
-* □ Medical
-* □ Others: ________________
-
-### TREATMENT TYPE
-* □ In Patient
-* □ Day Care
-* Transfer To: ________________
-
-### CONTACT DETAILS
-**Person to Contact (Next of Kin)**:
-* Name: ________________
-* Relationship with Patient: ________________
-* Address: ________________
-* Pincode: ________________
-* Mobile: ________________
-* Email: ________________
-
-### OFFICIAL USE
-* Front Office Executive Name:
-* Front Office Executive Signature:
-* Advance Amount Paid:
-* ICD-10 Code (For Medical Records Section):
-
-*\\* Subject to change during the course of diseases*
-
----
-*Form No: KMHIPF002V3*
-            """
-        },
-        {
-            "role": "user",
-            "content": [
-                {
-                    "type": "text",
-                    "text": "Extract and format the Patient Admission Form from this image according to the specified markdown format. Preserve all form fields and checkboxes (as □)."
-                },
-                {
-                    "type": "image_url",
-                    "image_url": {
-                        "url": f"data:image/png;base64,{img_base64}"
-                    }
-                }
-            ]
-        }
-    ],
-    max_tokens=4096
-)
-
-return response.choices[0].message.content
+                        # PATIENT ADMISSION FORM
+                        ## DR.KAMAKSHI MEMORIAL HOSPITAL, PALLIKARANAI, CHENNAI.
+                        
+                        ### PATIENT PROFILE
+                        *Please paste the sticker within the box*
+                        
+                        * UHID: ______
+                        * Patient Name: ______
+                        * Age/Gender: ______
+                        * Doctor Name: ______
+                        
+                        ### BASIC INFORMATION
+                        * Date & Time of Admission: [DD/MM/YYYY]
+                        * Date of Birth: [DD/MM/YYYY]
+                        
+                        ### IDENTIFICATION
+                        **ID Proof Already Registered**: □ Yes □ No
+                        
+                        **Type of ID**:
+                        * □ Aadhar
+                        * □ Passport
+                        * □ Voter ID
+                        * □ Driving License
+                        * □ Others
+                        
+                        ID No.: ________________
+                        Contact No.: ________________
+                        
+                        ### MEDICAL DETAILS
+                        * Provisional Diagnosis:
+                        * Reason for Admission:
+                        * Plan of Care:
+                        * Expected Outcome*:
+                        
+                        ### CONSULTANT DETAILS
+                        * Primary Consultant Name:
+                        * Speciality:
+                        
+                        ### PATIENT REFERENCE INFORMATION
+                        *(To be filled by Front Office)*
+                        
+                        **Reference Via**:
+                        * □ Doctor
+                        * □ Hospital
+                        * □ Ambulance
+                        * □ DRKMH Employee
+                        * □ Self / Walk In
+                        
+                        **Referrer Details**:
+                        * Name: ________________
+                        * Contact No.: ________________
+                        
+                        ### TYPE OF ADMISSION
+                        * □ Emergency
+                        * □ Elective
+                        * □ MLC
+                        * □ Surgery
+                        * □ Medical
+                        * □ Others: ________________
+                        
+                        ### TREATMENT TYPE
+                        * □ In Patient
+                        * □ Day Care
+                        * Transfer To: ________________
+                        
+                        ### CONTACT DETAILS
+                        **Person to Contact (Next of Kin)**:
+                        * Name: ________________
+                        * Relationship with Patient: ________________
+                        * Address: ________________
+                        * Pincode: ________________
+                        * Mobile: ________________
+                        * Email: ________________
+                        
+                        ### OFFICIAL USE
+                        * Front Office Executive Name:
+                        * Front Office Executive Signature:
+                        * Advance Amount Paid:
+                        * ICD-10 Code (For Medical Records Section):
+                        
+                        *\\* Subject to change during the course of diseases*
+                        
+                        ---
+                        *Form No: KMHIPF002V3*
+                                    """
+                            },
+                            {
+                                "role": "user",
+                                "content": [
+                                    {
+                                        "type": "text",
+                                        "text": "Extract and format the Patient Admission Form from this image according to the specified markdown format. Preserve all form fields and checkboxes (as □)."
+                                    },
+                                    {
+                                        "type": "image_url",
+                                        "image_url": {
+                                            "url": f"data:image/png;base64,{img_base64}"
+                                        }
+                                    }
+                                ]
+                            }
+                        ],
+                        max_tokens=4096
+                    )
+                    
+                    return response.choices[0].message.content
+                
+                except Exception as e:
+                print(f"Error in text extraction: {str(e)}")
+                return None
 
 def extract_text(pdf_file):
     if OPENAI_API_KEY is None: