Update app.py

app.py

@@ -27,56 +27,72 @@ def generate_code(summary, language):
         st.error(f"Error generating code: {e}")
         return ""
 
+# Function to explain the generated code using Llama
 def explain_code(code):
-  try:
-    chat_completion = client.chat.completions.create(
-        messages=[
-          {
-            "role": "user",
-            "content": f"Explain the following code descriptively and attractively so the user can easily understand it:\n\n{code}",
-          }
-        ],
-        model="llama3-8b-8192",
-        stream=False,
-    )
-    explanation = chat_completion.choices[0].message.content
-    return explanation
-  except Exception as e:
-    st.error(f"Error explaining code: {e}")
-    return ""
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "user",
+                    "content": f"Explain the following code descriptively and attractively so the user can easily understand it:\n\n{code}",
+                }
+            ],
+            model="llama3-8b-8192",
+            stream=False,
+        )
+        explanation = chat_completion.choices[0].message.content
+        return explanation
+    except Exception as e:
+        st.error(f"Error explaining code: {e}")
+        return ""
 
+# Function to save code as a PDF
 def save_code_as_pdf(code, file_name="generated_code.pdf"):
-  pdf = FPDF()
-  pdf.add_page()
-  pdf.set_font("Arial", size=12)
-  pdf.multi_cell(0, 10, code)
-  pdf.output(file_name)
-  return file_name
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+    pdf.multi_cell(0, 10, code)
+    pdf.output(file_name)
+    return file_name
 
-# ... rest of your code ...
+# --- Streamlit Interface ---
+st.set_page_config(page_title="Generative AI Code Generator", page_icon="🧑‍💻", layout="wide")
 
-if st.button("Generate Code"):
-  # ... your code to generate code ...
-  if generated_code:
-    # ... your code to display generated code ...
+# Page Title
+st.title("🚀 Generative AI Code Generator Using StarCoder")
 
-    # Explanation Button
-    if st.button("Explain Code"):
-      explanation = explain_code(generated_code)
-      st.subheader("📖 Code Explanation:")
-      st.write(explanation)
-
-    # Download Code as PDF
-    if st.button("Download Code as PDF"):
-      pdf_path = save_code_as_pdf(modified_code)  # Use modified_code or generated_code based on your preference
-      with open(pdf_path, "rb") as pdf_file:
-        st.download_button(
-            label="📥 Download PDF",
-            data=pdf_file,
-            file_name="generated_code.pdf",
-            mime="application/pdf",
-        )
+# Input Fields
+summary = st.text_area("📝 Enter the Task Summary", "For example: Create a function to add two numbers.")
+language = st.selectbox("🌐 Select Programming Language", ["Python", "Java", "JavaScript", "C++"])
 
+# Generate Code Button
+if st.button("Generate Code"):
+    if summary:
+        generated_code = generate_code(summary, language)
+
+        if generated_code:
+            st.subheader(f"✨ Generated {language} Code:")
+            st.code(generated_code, language=language.lower())
+
+            # Code Modification Section
+            modified_code = st.text_area("✏️ Modify the Code (Optional):", value=generated_code, height=200)
+
+            # Explanation Button
+            if st.button("Explain Code"):
+                explanation = explain_code(generated_code)  # Use generated_code for explanation
+                st.subheader("📖 Code Explanation:")
+                st.write(explanation)
+
+            # Download Code as PDF
+            if st.button("Download Code as PDF"):
+                pdf_path = save_code_as_pdf(modified_code)  # Use modified_code if edited
+                with open(pdf_path, "rb") as pdf_file:
+                    st.download_button(
+                        label="📥 Download PDF",
+                        data=pdf_file,
+                        file_name="generated_code.pdf",
+                        mime="application/pdf",
+                    )
 
 # New Code Button
 if st.button("Generate New Code"):
@@ -84,9 +100,4 @@ if st.button("Generate New Code"):
 
 # Footer Information
 st.markdown("---")
-st.write("🌟 Powered by **Streamlit**, **Groq**, and **StarCoder** | Deployed on Hugging Face")
-
-
-
-
-
+st.write("🌟 Powered by **Streamlit**, **Groq**, and **StarCoder** | Deployed on Hugging Face")