Create app.py

app.py

@@ -0,0 +1,26 @@
+import streamlit as st
+import requests
+
+API_URL = "http://127.0.0.1:8000"
+
+st.title("Mechanistic Analysis Interface")
+st.write("Run GPT-2 generation + activation patching experiments.")
+
+prompt = st.text_area("Enter your sentence:")
+
+if st.button("Run Experiment"):
+    if prompt:
+        with st.spinner("Running experiment..."):
+            response = requests.post(f"{API_URL}/generate", json={"prompt": prompt})
+            data = response.json()
+        
+        st.subheader("Generated Text")
+        st.write(data["generated_text"])
+        
+        st.subheader("Activation Patching Traces")
+        st.write(data["activations"])
+        
+        st.subheader("Explanation")
+        st.write(data["explanation"])
+        
+        st.success(f"Experiment saved with ID: {data['id']}")