Update src/streamlit_app.py

src/streamlit_app.py

@@ -171,16 +171,22 @@ elif choose == "CNN":
     training_file = st.file_uploader("Upload Data Training (.txt)", type=["txt"])
     real_files = st.file_uploader("Upload Data Real (.txt)", type=["txt"], accept_multiple_files=True)
 
-    if training_file is not None:
-        # Proses file training langsung dari memori
-        training_content = training_file.read().decode("utf-8")
-        st.write("Training file content:", training_content[:100])  # Contoh
-
+    if training_file:
+        if training_file.size > 50 * 1024 * 1024:  # Batasi ukuran file <= 50MB
+            st.error("Ukuran file training melebihi 50MB!")
+        else:
+            content = training_file.read().decode("utf-8")
+            st.write("File training berhasil diupload ✅")
+            st.code(f"100 karakter pertama:\n{content[:100]}")
+    
     if real_files:
         for file in real_files:
-            # Proses setiap file real langsung dari memori
-            real_content = file.read().decode("utf-8")
-            st.write(f"Content dari {file.name}:", real_content[:100])
+            if file.size > 50 * 1024 * 1024:
+                st.error(f"File {file.name} melebihi 50MB!")
+            else:
+                content = file.read().decode("utf-8")
+                st.write(f"File {file.name} berhasil diupload ✅")
+                st.code(f"100 karakter pertama:\n{content[:100]}")
 
     # Parameter model
     epochs = st.number_input("Jumlah Epoch", min_value=1, value=2000)