Update app.py

app.py

@@ -1,148 +1,71 @@
 import streamlit as st
-from streamlit.components.v1 import html
-
-
-state = st.session_state
-if 'my_value' not in state:
-    state['my_value'] = None
-if st.button('Set value'):
-    state['my_value'] = 'some_value'
-if state['my_value'] is not None:
-    st.write(f'<a href="?my_param={state["my_value"]}">Click here to see the value in the URL</a>')
+from pathlib import Path
+import json
+
+# 🦄 Magical state initialization
+if 'file_data' not in st.session_state:
+    st.session_state.file_data = {}
+
+def save_to_file(data, filename):
+    """🌈 Rainbow file saver - spreading joy through data persistence!"""
+    try:
+        with open(filename, 'w') as f:
+            json.dump(data, f)
+        return True
+    except Exception as e:
+        st.error(f"Oopsie! 🙈 Couldn't save to {filename}: {str(e)}")
+        return False
+
+def load_from_file(filename):
+    """🎀 Pretty file loader - bringing your data back with style!"""
+    try:
+        with open(filename, 'r') as f:
+            return json.load(f)
+    except FileNotFoundError:
+        st.info(f"✨ New adventure! No previous data found in {filename}")
+        return {}
+    except json.JSONDecodeError:
+        st.warning(f"🎭 Hmm, the file {filename} contains invalid JSON. Starting fresh!")
+        return {}
+
+def main():
+    st.title("🎉 Super Fun File Handler!")
     
-if state['my_value'] is not None:
-    st.markdown(state['my_value'])
+    # 🎨 Pretty file upload section
+    uploaded_file = st.file_uploader("Upload your happy little file!", type=['txt', 'json'])
     
-if state['my_value'] is not None:
-    st.write(f'<a href="?my_param={state["my_value"]}">Click here to see the value in the URL</a>')
-
-
-
+    if uploaded_file:
+        # 🎪 Process the uploaded file with circus-level excitement!
+        content = uploaded_file.getvalue().decode()
+        st.session_state.file_data[uploaded_file.name] = content
+        st.success(f"🌟 Woohoo! Successfully loaded {uploaded_file.name}")
     
-#from streamlit_cookies_controller import CookieController
-
-#st.set_page_config('Cookie QuickStart', '🍪', layout='wide')
-
-#controller = CookieController()
-
-# Set a cookie
-#controller.set('cookie_name', 'testing')
-
-# Get all cookies
-#cookies = controller.getAll()
-
-# Get a cookie
-#cookie = controller.get('cookie_name')
-
-# Remove a cookie
-#controller.remove('cookie_name')
-
-
-
-
-
-
-
-# Debugging Headers and Cookies
-# st.write("Headers:", st.context.headers)
-st.write("Cookies:", st.context.cookies)
-
-# Define JavaScript code
-javascript_code = """
-<script>
-    // Function to return a string
-    function myJavaScriptFunction() {
-        return "Hello from JavaScript!";
-    }
-
-    // Function to call and send data back to Python
-    function callAndCapture() {
-        const result = myJavaScriptFunction();
-        // Send the result back to Streamlit
-        window.parent.postMessage({ isStreamlitMessage: true, type: 'streamlit:js_return', data: result }, '*');
-    }
-
-    // Add an event listener to listen for messages from Streamlit
-    window.addEventListener('message', function(event) {
-        if (event.data && event.data.type === 'streamlit:buttonClick') {
-            callAndCapture();
-        }
-    });
-</script>
-<button onclick="callAndCapture()">Run JavaScript Function</button>
-"""
-
-# Embed JavaScript into Streamlit app
-html(javascript_code, height=300)
-
-# Placeholder to display results
-result_placeholder = st.empty()
-
-# Add Streamlit button to trigger JavaScript function
-if st.button("Call JavaScript Function"):
-    st.write("Waiting for JavaScript response...")
-    trigger_js = """
-    <script>
-    // Trigger the JavaScript function via a postMessage event
-    window.parent.postMessage({ type: 'streamlit:buttonClick' }, '*');
-    </script>
-    """
-    html(trigger_js, height=0)
-
-# Add a listener for JavaScript responses
-st.markdown(
-    """
-    <script>
-    // Listen for messages from JavaScript
-    window.addEventListener('message', (event) => {
-        if (event.data && event.data.type === 'streamlit:js_return') {
-            // Pass the data back to Streamlit as a new message
-            const result = event.data.data;
-            document.dispatchEvent(new CustomEvent('StreamlitResult', { detail: result }));
-        }
-    });
-
-    // Dispatch data from CustomEvent to Streamlit
-    document.addEventListener('StreamlitResult', (e) => {
-        const data = e.detail;
-        window.parent.postMessage({ type: 'streamlit:result', data: data }, '*');
-    });
-    </script>
-    """,
-    unsafe_allow_html=True,
-)
-
-# Use an iframe for communication
-if "js_result" not in st.session_state:
-    st.session_state["js_result"] = ""
-
-try:
-    # Listen for result data from JavaScript
-    received_result = html(
-        """
-        <script>
-        window.addEventListener('message', (event) => {
-            if (event.data && event.data.type === 'streamlit:result') {
-                const data = event.data.data;
-                // Send the result back to Streamlit
-                document.dispatchEvent(new CustomEvent('PythonReceivedResult', { detail: data }));
-            }
-        });
-
-        document.addEventListener('PythonReceivedResult', (e) => {
-            const received = e.detail;
-            document.getElementById('result-placeholder').innerText = received;
-        });
-        </script>
-        <div id="result-placeholder"></div>
-        """,
-        height=0,
-    )
-
-    # If there's a new result, update it in session state
-    if received_result and received_result != st.session_state["js_result"]:
-        st.session_state["js_result"] = received_result
-        result_placeholder.write(f"Result from JavaScript: {received_result}")
-
-except Exception as e:
-    st.error(f"An error occurred: {e}")
+    # 🎠 Display our magical collection of files
+    if st.session_state.file_data:
+        st.subheader("🗂️ Your Wonderful Files:")
+        for filename, content in st.session_state.file_data.items():
+            with st.expander(f"📁 {filename}"):
+                st.text_area("Content", content, height=150, key=filename)
+                
+                if st.button(f"💾 Save changes to {filename}"):
+                    updated_content = st.session_state[filename]
+                    if save_to_file(updated_content, filename):
+                        st.success(f"🎉 Yay! Saved {filename} successfully!")
+    
+    # 🎪 System prompt section
+    st.subheader("🤖 Super Smart System Prompt Generator")
+    if st.session_state.file_data:
+        selected_files = st.multiselect(
+            "Select files for your prompt",
+            list(st.session_state.file_data.keys())
+        )
+        
+        if selected_files:
+            prompt = "Here are the contents of the selected files:\n\n"
+            for file in selected_files:
+                prompt += f"File: {file}\n```\n{st.session_state.file_data[file]}\n```\n\n"
+            
+            st.text_area("Generated Prompt", prompt, height=300)
+
+if __name__ == "__main__":
+    main()