Update app.py

app.py

@@ -10,6 +10,7 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
 if 'generated_codes' not in st.session_state:
     st.session_state.generated_codes = []
 
+# Function to get the model pipeline
 @st.cache_resource
 def get_model_pipeline(model_name):
     try:
@@ -19,6 +20,7 @@ def get_model_pipeline(model_name):
         logging.error(f"Error loading model pipeline: {e}")
         return None
 
+# Function to generate code
 @st.cache_data
 def generate_code(task_description, max_length, temperature, num_return_sequences, model_name):
     code_pipeline = get_model_pipeline(model_name)
@@ -38,6 +40,7 @@ def generate_code(task_description, max_length, temperature, num_return_sequence
         logging.error(f"Error generating code: {e}")
         return [f"Error generating code: {e}"]
 
+# Function to save code to a file
 def save_code(code, file_name):
     try:
         with open(file_name, "w") as file:
@@ -47,6 +50,7 @@ def save_code(code, file_name):
         logging.error(f"Error saving code: {e}")
         return False
 
+# Main function for Streamlit app
 def main():
     st.set_page_config(page_title="Advanced Code Generator", layout="wide")
 
@@ -70,7 +74,7 @@ def main():
     st.header("Options")
     col1, col2, col3 = st.columns(3)
     with col1:
-        max_length = st.slider("Max Length", min_value=50, max_value=1000, value=250, step=50, help="Maximum length of the generated code.")
+        max_length = st.slider("Max Length", min_value=50, max_value=2048, value=250, step=50, help="Maximum length of the generated code.")
     with col2:
         temperature = st.slider("Temperature", min_value=0.1, max_value=1.0, value=0.7, step=0.1, help="Controls the creativity of the generated code.")
     with col3:
@@ -79,6 +83,8 @@ def main():
     # Generate Code Button
     if st.button("Generate Code"):
         if task_description.strip():
+            # Clear previous generated codes
+            st.session_state.generated_codes = []
             with st.spinner("Generating code..."):
                 st.session_state.generated_codes = generate_code(task_description, max_length, temperature, num_return_sequences, model_name)
             st.header("Generated Code")