removed comments in final render and added _FILES variable

prodev717 · prodev717 · commit dda503319eb6 · 2025-07-03T18:36:59.000+05:30
diff --git a/.gitignore b/.gitignore
@@ -8,3 +8,5 @@ wheels/
 
 # Virtual environments
 .venv
+
+models/
diff --git a/README.md b/README.md
@@ -11,7 +11,7 @@
 * ✅ Built with Flask under the hood
 * ✅ Minimal setup — no frontend frameworks or build tools needed
 * ✅ Ideal for quick demos, server-rendered sites, teaching, and prototyping AI apps
-* ✅ Supports special variables like `_GET`, `_POST`, `GLOBAL`, and `SESSION` inside `<wep>` blocks for handling requests and session data
+* ✅ Supports special variables like `_GET`, `_POST`, `_FILES`, `GLOBAL`, and `SESSION` inside `<wep>` blocks for handling requests and session data
 
 
 ---
diff --git a/load.py b/load.py
@@ -0,0 +1,12 @@
+from transformers import BlipProcessor, BlipForConditionalGeneration
+
+# to save model locally for Image Caption Generator 
+
+model = "Salesforce/blip-image-captioning-base"
+save = "./models/blip"
+
+processor = BlipProcessor.from_pretrained(model)
+model = BlipForConditionalGeneration.from_pretrained(model)
+
+processor.save_pretrained(save)
+model.save_pretrained(save)
diff --git a/main.py b/main.py
@@ -24,6 +24,7 @@ def echo(tag):
             "echo" : echo,
             "_GET" : request.args.to_dict(),
             "_POST" : request.form.to_dict(),
+            "_FILES": request.files.to_dict(),
             "GLOBAL" : GLOBAL,
             "SESSION" : session
         }
@@ -36,9 +37,10 @@ def echo(tag):
             return f"<pre style='background-color:#7f1d1d; color:#fca5a5; padding:1rem; border-radius:0.5rem;'>Execution Error:\n{error}</pre>"
     try:
         processed_html = re.sub(r"<wep>(.*?)</wep>", replace_wep, html, flags=re.DOTALL)
+        comments_removed = re.sub("(<!--.*?-->)","",processed_html,flags=re.DOTALL)
     except re.error as re_err:
         return Response(f"Regex Error: {re_err}", status=500, mimetype="text/plain")
-    return Response(processed_html, mimetype="text/html; charset=utf-8")
+    return Response(comments_removed, mimetype="text/html; charset=utf-8")
 
 app = Flask(__name__)
 app.secret_key = 'web-embedded-python'
diff --git a/public/caption.wep b/public/caption.wep
@@ -0,0 +1,72 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <title>Image Caption Generator Demo — WEP</title>
+    <script src="https://cdn.tailwindcss.com"></script>
+    <script>
+      tailwind.config = {
+        darkMode: 'class'
+      }
+    </script>
+</head>
+<body class="dark bg-gray-900 text-gray-100 min-h-screen flex flex-col items-center justify-start py-10 px-4">
+
+    <div class="w-full max-w-xl bg-gray-800 shadow-lg rounded-lg p-6">
+        <h1 class="text-2xl font-bold mb-6 text-center text-teal-400">🖼️ Image Caption Generator</h1>
+
+        <form method="POST" enctype="multipart/form-data" class="flex flex-col items-center gap-4">
+            <input type="file" name="myfile" accept="image/*" required
+                class="block w-full text-sm text-gray-300
+                       file:mr-4 file:py-2 file:px-4
+                       file:rounded-md file:border-0
+                       file:text-sm file:font-semibold
+                       file:bg-teal-100 file:text-teal-900
+                       hover:file:bg-teal-200" />
+            <button type="submit"
+                class="bg-teal-500 text-white px-6 py-2 rounded-md hover:bg-teal-600 transition">
+                Generate Caption
+            </button>
+        </form>
+
+        <div class="mt-8">
+            <wep>
+import io
+import base64
+from PIL import Image
+from transformers import BlipProcessor, BlipForConditionalGeneration
+
+processor = BlipProcessor.from_pretrained("./models/blip")
+model = BlipForConditionalGeneration.from_pretrained("./models/blip")
+
+if "myfile" in _FILES:
+    file = _FILES["myfile"]
+    try:
+        img_bytes = file.read()
+        image = Image.open(io.BytesIO(img_bytes)).convert("RGB")
+
+        inputs = processor(images=image, return_tensors="pt")
+        output = model.generate(**inputs)
+        caption = processor.decode(output[0], skip_special_tokens=True)
+
+        b64img = base64.b64encode(img_bytes).decode('utf-8')
+
+        echo(f'''
+            <div class="text-center">
+                <h3 class="text-lg font-semibold text-gray-200 mb-2">Caption:</h3>
+                <p class="text-xl text-green-300 mb-4">{caption}</p>
+                <img src="data:image/jpeg;base64,{b64img}" class="mx-auto rounded-lg shadow-lg max-w-xs border border-gray-600" />
+            </div>
+        ''')
+    except Exception as e:
+        echo(f'<div class="text-red-400"><h3 class="font-bold">Something went wrong</h3><pre>{e}</pre></div>')
+else:
+    echo("<p class='text-center text-gray-400'>No file uploaded yet.</p>")
+            </wep>
+        </div>
+    </div>
+    <footer class="pt-8 text-center text-gray-500 text-sm">
+      Built with ❤️ using WEP — Web Embedded Python
+    </footer>
+</body>
+</html>
diff --git a/public/index.wep b/public/index.wep
@@ -182,9 +182,14 @@ else:
       </div>
     </section>
 <!-- Section 7: Additional Link -->
-    <a href="/aidemo.wep">
-      <h2 class="text-xl font-semibold text-teal-300 hover:underline">AI app demo</h2>
+    <section>
+    <a href="/lrdemo.wep">
+      <h2 class="text-xl font-semibold text-teal-300 hover:underline">Linear Regression Demo</h2>
+    </a>
+    <a href="/caption.wep">
+      <h2 class="text-xl font-semibold text-teal-300 hover:underline">Image Caption Generator Demo</h2>
     </a>
+    </section>
     <!-- Footer -->
     <footer class="text-center text-gray-600 pt-12">
       <hr class="mb-4 border-gray-700" />
diff --git a/public/lrdemo.wep b/public/lrdemo.wep
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,4 +8,7 @@ dependencies = [
     "flask>=3.1.1",
     "matplotlib>=3.10.3",
     "numpy>=2.2.6",
+    "pillow>=11.2.1",
+    "torch>=2.7.1",
+    "transformers>=4.53.0",
 ]
diff --git a/uv.lock b/uv.lock

-Original file line number
+Diff line change
 # Virtual environments
 .venv
++
 +models/
Original file line number	Diff line number	Diff line change
`@@ -8,4 +8,7 @@ dependencies = [`
`8`	`8`	`"flask>=3.1.1",`
`9`	`9`	`"matplotlib>=3.10.3",`
`10`	`10`	`"numpy>=2.2.6",`
	`11`	`+ "pillow>=11.2.1",`
	`12`	`+ "torch>=2.7.1",`
	`13`	`+ "transformers>=4.53.0",`
`11`	`14`	`]`