Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,154 +1,57 @@
-import streamlit as st
-import json
-from pathlib import Path
-
-st.set_page_config(page_title="Shader + Comments JSON", layout="wide")
-
-st.title("🌀 Shader + Comments (Hugging Face Ready)")
-
-# -------------------------------
-# Use /data folder for persistent storage
-# -------------------------------
-DATA_DIR = Path("./data")
-DATA_DIR.mkdir(exist_ok=True)
-COMMENTS_FILE = DATA_DIR / "comments.json"
-
-# Load existing comments
-if COMMENTS_FILE.exists():
-    with open(COMMENTS_FILE, "r") as f:
-        comments = json.load(f)
-else:
-    comments = []
-
-# -------------------------------
-# Determine current shader
-# -------------------------------
-shader_code_default = """
-precision mediump float;
-uniform vec2 iResolution;
-uniform float iTime;
-
-void mainImage(out vec4 fragColor, in vec2 fragCoord) {
-    vec2 uv = fragCoord / iResolution.xy;
-    vec3 col = 0.5 + 0.5*cos(iTime + uv.xyx + vec3(0,2,4));
-    fragColor = vec4(col,1.0);
-}
-void main() { mainImage(gl_FragColor, gl_FragCoord.xy); }
-"""
-
-latest_shader = shader_code_default
-for c in reversed(comments):
-    if c.get("type") == "shader":
-        latest_shader = c["content"]
-        break
-
-# -------------------------------
-# Shader editor
-# -------------------------------
-shader_code = st.text_area("Shader Code (GLSL Fragment Shader)", latest_shader, height=250)
-
-# -------------------------------
-# WebGL Canvas
-# -------------------------------
-html_code = f"""
-<canvas id="glcanvas" width="800" height="500"></canvas>
-<script>
-const canvas = document.getElementById('glcanvas');
-const gl = canvas.getContext('webgl');
-if (!gl) {{
-  alert('WebGL not supported');
-}}
-
-const vertCode = `
-attribute vec4 a_position;
-void main() {{
-  gl_Position = a_position;
-}}
-`;
-
-const fragCode = `{shader_code}`;
-
-function createShader(gl, type, source) {{
-  const shader = gl.createShader(type);
-  gl.shaderSource(shader, source);
-  gl.compileShader(shader);
-  if (!gl.getShaderParameter(shader, gl.COMPILE_STATUS)) {{
-    alert(gl.getShaderInfoLog(shader));
-  }}
-  return shader;
-}}
-
-const vertexShader = createShader(gl, gl.VERTEX_SHADER, vertCode);
-const fragmentShader = createShader(gl, gl.FRAGMENT_SHADER, fragCode);
-
-const program = gl.createProgram();
-gl.attachShader(program, vertexShader);
-gl.attachShader(program, fragmentShader);
-gl.linkProgram(program);
-gl.useProgram(program);
-
-const positionBuffer = gl.createBuffer();
-gl.bindBuffer(gl.ARRAY_BUFFER, positionBuffer);
-gl.bufferData(gl.ARRAY_BUFFER, new Float32Array([
-  -1, -1, 1, -1, -1, 1,
-  -1, 1, 1, -1, 1, 1
-]), gl.STATIC_DRAW);
-
-const positionLoc = gl.getAttribLocation(program, "a_position");
-gl.enableVertexAttribArray(positionLoc);
-gl.vertexAttribPointer(positionLoc, 2, gl.FLOAT, false, 0, 0);
-
-const iResolution = gl.getUniformLocation(program, "iResolution");
-const iTime = gl.getUniformLocation(program, "iTime");
-
-function render(time) {{
-  gl.uniform2f(iResolution, canvas.width, canvas.height);
-  gl.uniform1f(iTime, time * 0.001);
-  gl.drawArrays(gl.TRIANGLES, 0, 6);
-  requestAnimationFrame(render);
-}}
-
-requestAnimationFrame(render);
-</script>
-"""
-
-st.components.v1.html(html_code, height=520)
-
-# -------------------------------
-# Comments Section
-# -------------------------------
-st.markdown("---")
-st.subheader("💬 Comments / Add New Shader")
-
-comment_type = st.radio("Type", ["Normal Comment", "New Shader"])
-
-with st.form("comment_form", clear_on_submit=True):
-    user_name = st.text_input("Your name", "")
-    user_content = st.text_area("Content", "", height=80)
-    submitted = st.form_submit_button("Submit")
-
-    if submitted and user_content.strip():
-        new_comment = {
-            "name": user_name or "Anonymous",
-            "content": user_content,
-            "type": "shader" if comment_type == "New Shader" else "comment"
-        }
-        comments.append(new_comment)
-
-        # Save to JSON in /data folder
-        with open(COMMENTS_FILE, "w") as f:
-            json.dump(comments, f, indent=2)
-
-        st.success("Saved!")
-
-        # If it's a shader, update editor immediately
-        if new_comment["type"] == "shader":
-            st.experimental_rerun()
-
-# Display all comments
-for c in reversed(comments):
-    if c["type"] == "shader":
-        st.markdown(f"**{c['name']} added a new shader:**")
-        st.code(c["content"], language="glsl")
-    else:
-        st.markdown(f"**{c['name']}**: {c['content']}")
+import streamlit as st
+import json
+import os
+
+# --- Paths ---
+COMMENTS_FILE = "data/comments.json"
+os.makedirs("data", exist_ok=True)
+
+# --- Load existing comments ---
+if os.path.exists(COMMENTS_FILE):
+    with open(COMMENTS_FILE, "r") as f:
+        comments = json.load(f)
+else:
+    comments = []
+
+# --- Shader Editor ---
+st.subheader("Shader Editor")
+default_shader = """// Fresh GLSL shader example
+#ifdef GL_ES
+precision mediump float;
+#endif
+
+uniform float iTime;      
+uniform vec2 iResolution; 
+
+void main() {
+    vec2 uv = gl_FragCoord.xy / iResolution.xy;
+    vec3 color = 0.5 + 0.5 * cos(iTime + uv.xyx * 3.0);
+    gl_FragColor = vec4(color, 1.0);
+}
+"""
+shader_code = st.text_area("Edit shader code", default_shader, height=300)
+
+# --- Comments Section ---
+st.subheader("Comments")
+if comments:
+    for c in comments:
+        st.write(f"**{c['name']}**: {c['content']}")
+else:
+    st.write("_No comments yet_")
+
+# --- Add a new comment ---
+with st.form("comment_form", clear_on_submit=True):
+    user_name = st.text_input("Your name", "")
+    user_content = st.text_area("Content", height=80)
+    submitted = st.form_submit_button("Submit")
+
+    if submitted and user_content.strip():
+        new_comment = {
+            "name": user_name or "Anonymous",
+            "content": user_content
+        }
+        comments.append(new_comment)
+        with open(COMMENTS_FILE, "w") as f:
+            json.dump(comments, f, indent=2)
+        st.success("Saved!")
+        st.experimental_rerun()  # refresh to show new comment immediately