chore: App file generated

app.py

@@ -0,0 +1,61 @@
+import streamlit as st
+import streamlit.components.v1 as components
+import subprocess
+import os
+import glob
+
+# Function to run the model and generate HTML
+def generate_html(text_input):
+    command = [
+        "python", "generate.py",
+        "--resume-pth", "output/vq/2023-07-19-04-17-17_12_VQVAE_20batchResetNRandom_8192_32/net_last.pth",
+        "--resume-trans", "output/t2m/2023-10-10-03-17-01_HML3D_44_crsAtt2lyr_mask0.5-1/net_last.pth",
+        "--text", text_input,
+        "--length", str(st.session_state.length)
+    ]
+    # Run the command
+    subprocess.run(command, check=True)
+    # Return the path of the latest HTML file in the output directory
+    return find_latest_html_file('output')
+
+# Function to find the latest HTML file in the specified directory
+def find_latest_html_file(base_path):
+    # List all HTML files in the output directory and find the latest one
+    list_of_files = glob.glob(f'{base_path}/*.html')
+    if not list_of_files:
+        return None
+    latest_file = max(list_of_files, key=os.path.getctime)
+    return latest_file
+
+# Streamlit app
+components.html("""
+    <h1 style='text-align: center; color: white;'>MMM Model Demo</h1>
+    """,
+    height=100)
+
+# Initialize session state variable for text input
+text_input = st.text_area("Enter text here:", value="", key="text_input")
+length = st.number_input("Length of the generated text:", value=156, key="length")
+
+# Store the input text in session state
+st.session_state.input_text = text_input
+st.session_state.length = length
+
+# Button trigger to generate HTML
+if st.button("Generate HTML"):
+    st.session_state.generate_html = True
+else:
+    st.session_state.generate_html = False
+
+# Check if HTML generation was triggered and input is not empty
+if st.session_state.text_input and st.session_state.generate_html:
+    html_file_path = generate_html(st.session_state.text_input)  # Uses session state
+    if html_file_path and os.path.exists(html_file_path):
+        with open(html_file_path, 'r') as file:
+            html_content = file.read()
+        
+        # Embed HTML content in an iframe
+        iframe = f"<iframe srcdoc='{html_content}' style='width:100%; height:800px; border:none;'></iframe>"
+        components.html(iframe, height=800)
+    else:
+        st.error("Generated HTML file not found.")