Update app.py

app.py

@@ -1,67 +1,69 @@
-import streamlit as st
-from transformers import pipeline, set_seed
-import logging
+ import streamlit as st
+   from transformers import pipeline, GPT2LMHeadModel, GPT2Tokenizer
+   import logging
 
-# Logging Setup
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+   # Logging Setup
+   logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-# Initialize the generator with seed for reproducibility
-set_seed(42)
-generator = pipeline('text-generation', model='bigscience/T0_3B')
+   # Initialize the generator with GPT-2
+   model_name = "gpt2"
+   model = GPT2LMHeadModel.from_pretrained(model_name)
+   tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+   generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
 
-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}"
-        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 codes
-    except Exception as e:
-        logging.error(f"Error generating code: {e}")
-        return [f"Error generating code: {e}"]
+   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}"
+           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 codes
+       except Exception as e:
+           logging.error(f"Error generating code: {e}")
+           return [f"Error generating code: {e}"]
 
-def main():
-    st.set_page_config(page_title="Advanced Code Generator", layout="wide")
+   def main():
+       st.set_page_config(page_title="Advanced Code Generator", layout="wide")
 
-    st.title("Advanced Code Generator")
-    st.markdown("This application generates code based on the given task description using a text-generation model.")
+       st.title("Advanced Code Generator")
+       st.markdown("This application generates code based on the given task description using a text-generation model.")
 
-    # Input Section
-    st.header("Task Description")
-    task_description = st.text_area("Describe the task for which you need code:", height=150)
+       # Input Section
+       st.header("Task Description")
+       task_description = st.text_area("Describe the task for which you need code:", height=150)
 
-    # 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.")
+       # 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.")
 
-    # 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.")
+       # 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")
-        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.")
+       # 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")
+          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()
+   if __name__ == "__main__":
+       main()