Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 import streamlit as st
 import google.generativeai as genai
 import requests
-import json
 
 # Configure the Gemini API
 genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
@@ -19,7 +18,7 @@ generation_config = {
 model = genai.GenerativeModel(
     model_name="gemini-1.5-pro",
     generation_config=generation_config,
-    system_instruction="You are a human named Ath who is a highly knowledgeable code assistant. You speak in a friendly and casual tone, using bro words, and you never mention you're an AI. Your responses should contain code only, without explanations.",
+    system_instruction="You are a human named Ath who is a highly knowledgeable code assistant. You speak in a friendly and casual tone, using bro words, and you never mention you're an AI. Your responses should include HTML, CSS, and JavaScript code separately when applicable, with clear section headers for each type of code.",
 )
 chat_session = model.start_chat(history=[])
 
@@ -56,10 +55,21 @@ if st.button("Generate Code"):
             completed_text = generate_response(prompt)
             st.success("Code generated successfully!")
             
-            # Split the response into separate code blocks based on a custom delimiter
-            code_blocks = completed_text.split("\n\n")
-            for block in code_blocks:
-                st.code(block, language="python")
+            # Display code blocks based on predefined markers
+            if "HTML:" in completed_text:
+                html_code = completed_text.split("HTML:")[1].split("CSS:")[0].strip()
+                st.subheader("HTML Code")
+                st.code(html_code, language="html")
+
+            if "CSS:" in completed_text:
+                css_code = completed_text.split("CSS:")[1].split("JavaScript:")[0].strip()
+                st.subheader("CSS Code")
+                st.code(css_code, language="css")
+
+            if "JavaScript:" in completed_text:
+                js_code = completed_text.split("JavaScript:")[1].strip()
+                st.subheader("JavaScript Code")
+                st.code(js_code, language="javascript")
 
 st.markdown("""
 <style>