Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from main import *
 
 # Set the title of the web app
 st.title("Text and Token Input App")
@@ -9,17 +10,26 @@ prompt = st.text_area("Enter a prompt above 15 words:")
 # Input for the number of tokens to be generated
 num_tokens = st.number_input("Enter the number of tokens to be generated:", min_value=1, step=1)
 
-# Function to calculate the length of the string and the sum of the length and number of tokens
-def calculate_length_and_sum(prompt, num_tokens):
-    prompt_length = len(prompt.split())
-    total_sum = prompt_length + num_tokens
-    return prompt_length, total_sum
+
+# Function to generate output based on the prompt and number of tokens
+def generate(prompt, num_tokens):
+    output = gen(prompt, num_tokens)
+    return output
+
 
 # Button to perform the calculation
-if st.button("Calculate"):
-    if len(prompt.split()) < 15:
+if st.button("Generate Output"):
+    if len(prompt.split()) < 9:
         st.warning("Please enter a prompt with more than 15 words.")
     else:
-        prompt_length, total_sum = calculate_length_and_sum(prompt, num_tokens)
-        st.write(f"Length of the prompt: {prompt_length}")
-        st.write(f"Sum of the length of the prompt and the number of tokens: {total_sum}")
+        output = generate(prompt, num_tokens)
+
+        # Display prompt length and sum of prompt length and number of tokens
+        prompt_length = len(prompt.split())
+        total_sum = prompt_length + num_tokens
+
+        # Display output in a box below or next to the input prompts
+
+
+        st.subheader("Generated Output")
+        st.write(output)