Update app.py

app.py

@@ -7,7 +7,7 @@ st.title("AI Code Bot")
 # Introduction
 st.write("""
 ### Generate Code Snippets with AI
-Enter a programming-related prompt, and the AI will generate a code snippet for you.
+Enter a programming-related prompt, and the AI will generate a code snippet with explanatory comments.
 """)
 
 # Input box for user query
@@ -33,10 +33,23 @@ if st.button("Generate Code"):
         with st.spinner("Generating code..."):
             try:
                 # Generate code using the model
-                result = model(user_input, max_length=100, num_return_sequences=1)
-                generated_code = result[0]["generated_text"]
+                result = model(user_input, max_length=200, num_return_sequences=1)
+                generated_text = result[0]["generated_text"]
                 
-                # Display the output
-                st.code(generated_code, language="python")
+                # Split comments and code for better presentation
+                lines = generated_text.split("\n")
+                comments = [line for line in lines if line.strip().startswith("#")]
+                code = [line for line in lines if not line.strip().startswith("#")]
+
+                # Display comments as markdown
+                if comments:
+                    st.subheader("Comments")
+                    for comment in comments:
+                        st.markdown(comment)
+
+                # Display code as Python syntax-highlighted block
+                if code:
+                    st.subheader("Code")
+                    st.code("\n".join(code), language="python")
             except Exception as e:
                 st.error(f"An error occurred: {e}")