Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,11 +6,12 @@ from pathlib import Path
 # SETUP
 # ============================================================================
 
-# Create logs folder
 LOGS_DIR = Path("logs")
+DATA_DIR = Path("data")
+
 LOGS_DIR.mkdir(exist_ok=True)
+DATA_DIR.mkdir(exist_ok=True)
 
-# Setup logging
 logging.basicConfig(
     level=logging.DEBUG,
     format='%(asctime)s - %(levelname)s - %(message)s',
@@ -24,27 +25,49 @@ logger = logging.getLogger(__name__)
 st.set_page_config(page_title="Math AI", layout="wide")
 
 # ============================================================================
-# STEP 1: BASIC LOGGING TEST
+# STEP 2: FILE UPLOAD
 # ============================================================================
 
 st.title("🧮 Math AI System")
-st.markdown("### Step 1: Logging System")
-
-logger.info("App loaded successfully")
-
-st.success("✅ Logging is working!")
-
-# Test logging
-if st.button("Click to test logger"):
-    logger.debug("User clicked the button")
-    logger.info("This is an info message")
-    st.write("Check logs below ↓")
-
-# Show logs
-if st.checkbox("Show logs"):
-    log_file = LOGS_DIR / "app.log"
-    if log_file.exists():
-        with open(log_file, 'r') as f:
-            st.code(f.read())
-    else:
-        st.info("No logs yet")
+st.markdown("### Step 2: File Upload")
+
+logger.info("App loaded")
+
+# File upload
+st.subheader("Upload a text file")
+
+uploaded_file = st.file_uploader(
+    "Choose a text file",
+    type=["txt", "json"]
+)
+
+if uploaded_file:
+    logger.info(f"File uploaded: {uploaded_file.name}")
+    
+    # Read file
+    content = uploaded_file.read().decode('utf-8')
+    
+    st.success(f"✅ File uploaded: {uploaded_file.name}")
+    st.write(f"Size: {len(content)} characters")
+    
+    # Show preview
+    with st.expander("📄 Preview (first 300 chars)"):
+        st.text(content[:300])
+    
+    # Save to data folder
+    if st.button("💾 Save to data folder"):
+        save_path = DATA_DIR / uploaded_file.name
+        with open(save_path, 'w') as f:
+            f.write(content)
+        logger.info(f"File saved to {save_path}")
+        st.success(f"✅ Saved to data/{uploaded_file.name}")
+
+# Show files in data folder
+st.subheader("📁 Files in data folder")
+files = list(DATA_DIR.glob("*"))
+if files:
+    for f in files:
+        size_kb = f.stat().st_size / 1024
+        st.write(f"- {f.name} ({size_kb:.1f} KB)")
+else:
+    st.info("No files yet. Upload one above!")