Update app.py

app.py

@@ -1,46 +1,67 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline, set_seed
 import logging
 
 # Logging Setup
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-# Initialize the generator
+# Initialize the generator with seed for reproducibility
+set_seed(42)
 generator = pipeline('text-generation', model='bigscience/T0_3B')
 
-def generate_code(task_description, max_length, temperature):
+def generate_code(task_description, max_length, temperature, num_return_sequences):
+    """Generate code based on the provided task description."""
     try:
         logging.info(f"Generating code with input: {task_description}")
         prompt = f"Develop code for the following task: {task_description}"
-        code = generator(prompt, max_length=max_length, num_return_sequences=1, temperature=temperature)[0]['generated_text'].strip()
+        response = generator(prompt, max_length=max_length, num_return_sequences=num_return_sequences, temperature=temperature)
+        codes = [resp['generated_text'].strip() for resp in response]
         logging.info("Code generation completed successfully.")
-        return code
+        return codes
     except Exception as e:
         logging.error(f"Error generating code: {e}")
-        return f"Error generating code: {e}"
+        return [f"Error generating code: {e}"]
 
-st.title("Advanced Code Generator")
+def main():
+    st.set_page_config(page_title="Advanced Code Generator", layout="wide")
 
-# Input Section
-st.header("Task Description")
-task_description = st.text_area("Describe the task for which you need code:", height=150)
+    st.title("Advanced Code Generator")
+    st.markdown("This application generates code based on the given task description using a text-generation model.")
 
-# Options Section
-st.header("Options")
-max_length = st.slider("Max Length", min_value=50, max_value=1000, value=250, step=50)
-temperature = st.slider("Temperature", min_value=0.0, max_value=1.0, value=0.7, step=0.1)
+    # Input Section
+    st.header("Task Description")
+    task_description = st.text_area("Describe the task for which you need code:", height=150)
 
-# Generate Code Button
-if st.button("Generate Code"):
-    if task_description:
-        generated_code = generate_code(task_description, max_length, temperature)
-        st.header("Generated Code")
-        st.code(generated_code, language='python')
+    # Options Section
+    st.header("Options")
+    max_length = st.slider("Max Length", min_value=50, max_value=1000, value=250, step=50, help="Maximum length of the generated code.")
+    temperature = st.slider("Temperature", min_value=0.0, max_value=1.0, value=0.7, step=0.1, help="Controls the creativity of the generated code.")
+    num_return_sequences = st.slider("Number of Sequences", min_value=1, max_value=5, value=1, step=1, help="Number of code snippets to generate.")
 
-# Save Code Section (Optional)
-if 'generated_code' in locals():
-    if st.button("Save Code"):
+    # Generate Code Button
+    if st.button("Generate Code"):
+        if task_description.strip():
+            with st.spinner("Generating code..."):
+                generated_codes = generate_code(task_description, max_length, temperature, num_return_sequences)
+            st.header("Generated Code")
+            for idx, code in enumerate(generated_codes):
+                st.subheader(f"Generated Code {idx + 1}")
+                st.code(code, language='python')
+        else:
+            st.error("Please enter a task description.")
+
+    # Save Code Section
+    if 'generated_codes' in locals() and generated_codes:
+        st.header("Save Code")
+        selected_code_idx = st.selectbox("Select which code to save:", range(1, len(generated_codes) + 1)) - 1
         file_name = st.text_input("Enter file name to save:", value="generated_code.py")
-        with open(file_name, "w") as file:
-            file.write(generated_code)
-        st.success(f"Code saved to {file_name}")
+        if st.button("Save", key="save_code"):
+            if file_name:
+                with open(file_name, "w") as file:
+                    file.write(generated_codes[selected_code_idx])
+                st.success(f"Code saved to {file_name}")
+            else:
+                st.error("Please enter a valid file name.")
+
+if __name__ == "__main__":
+    main()