Update app.py

app.py

@@ -1,10 +1,15 @@
 import base64
+import os
 from datetime import datetime
 from openai import OpenAI
 import gradio as gr
 
-# === Initialize OpenAI Client ===
-client = OpenAI()
+# === Initialize OpenAI Client using Environment Variable ===
+openai_api_key = os.environ.get("OPENAI_API_KEY")
+if not openai_api_key:
+    raise ValueError("OPENAI_API_KEY environment variable is not set.")
+
+client = OpenAI(api_key=openai_api_key)
 
 # === Prompts ===
 system_prompt = (
@@ -28,37 +33,48 @@ def encode_image_to_base64(image_file):
     return base64.b64encode(image_bytes).decode("utf-8")
 
 # === Transcription function ===
-def transcribe_image(image):
-    if image is None:
-        return "No image uploaded."
+def transcribe_images(images):
+    if not images:
+        return "No images uploaded."
+
+    results = []
+    for image in images:
+        encoded_image = encode_image_to_base64(image)
+        image_url = f"data:image/jpeg;base64,{encoded_image}"
 
-    encoded_image = encode_image_to_base64(image)
-    image_url = f"data:image/jpeg;base64,{encoded_image}"
+        response = client.chat.completions.create(
+            model="gpt-4-turbo",
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": [
+                    {"type": "text", "text": user_prompt_template},
+                    {"type": "image_url", "image_url": {"url": image_url}}
+                ]}
+            ],
+            max_tokens=1500
+        )
 
-    response = client.chat.completions.create(
-        model="gpt-4-turbo",
-        messages=[
-            {"role": "system", "content": system_prompt},
-            {"role": "user", "content": [
-                {"type": "text", "text": user_prompt_template},
-                {"type": "image_url", "image_url": {"url": image_url}}
-            ]}
-        ],
-        max_tokens=1500
-    )
+        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        result_text = f"🗓️ Transcribed on: {timestamp}\n\n{response.choices[0].message.content}"
+        results.append(result_text)
 
-    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    return f"🗓️ Transcribed on: {timestamp}\n\n{response.choices[0].message.content}"
+    return "\n\n---\n\n".join(results)
 
 # === Gradio Interface ===
 app = gr.Interface(
-    fn=transcribe_image,
-    inputs=gr.Image(type="file", label="Upload handwritten note (image)"),
-    outputs=gr.Textbox(label="Transcribed Output"),
+    fn=transcribe_images,
+    inputs=gr.File(
+        type="file",
+        label="Upload handwritten note images",
+        file_types=[".jpg", ".jpeg", ".png"],
+        multiple=True
+    ),
+    outputs=gr.Textbox(label="Transcribed Output", lines=30),
     title="Handwritten Note Transcriber",
-    description="Upload an image of handwritten notes to receive a clean, professional transcription."
+    description="Upload one or more images of handwritten notes to receive clean, professional transcriptions."
 )
 
 # === Launch ===
 if __name__ == "__main__":
     app.launch()
+