Update app.py

app.py

@@ -1,69 +1,69 @@
- import streamlit as st
-   from transformers import pipeline, GPT2LMHeadModel, GPT2Tokenizer
-   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 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)
+# 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")
-          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")
+        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()
+if __name__ == "__main__":
+    main()