Update app.py

app.py

@@ -27,7 +27,7 @@ def get_vectorstore():
 
 @st.cache_resource
 def get_memory():
-    """Initialize memory to store previous interactions."""
+    """Initialize memory for conversational interactions."""
     return ConversationBufferMemory(memory_key="chat_history", return_messages=True)
 
 # Initialize models and memory
@@ -35,53 +35,52 @@ llm = get_openai_model()
 vectorstore = get_vectorstore()
 memory = get_memory()
 
-# ------------------------------- PROMPT TEMPLATES -------------------------------
-prompt_template = PromptTemplate(
-    input_variables=["language", "description", "context"],
+# ------------------------------- STRONGER PROMPT TEMPLATES -------------------------------
+code_prompt = PromptTemplate(
+    input_variables=["language", "description"],
     template="""
-    You are an **expert AI coding assistant** with memory.
+    You are an **expert AI coding assistant** specializing in **ALL programming languages**.
 
-    **Context from previous interactions:**
-    {context}
-
-    **Current Task:** Generate a structured and functional code snippet.
+    **Task:** Generate an optimized, structured, and functional code snippet.
 
     **Language:** {language}  
-    **Description:** {description}  
+    **User Request:** {description}  
 
-    - Ensure the code follows best practices.
-    - Include comments explaining key sections.
-    - If applicable, provide installation/setup instructions.
+    - Follow industry best practices and clean code principles.
+    - Include relevant comments for clarity.
+    - If applicable, provide setup instructions.
+    - Use efficient algorithms and optimize performance.
+    - Ensure security measures are implemented where needed.
     """
 )
 
-enhance_template = PromptTemplate(
-    input_variables=["code", "context"],
+enhance_prompt = PromptTemplate(
+    input_variables=["code"],
     template="""
-    Improve the given code based on context:
-
-    **Previous Conversation Context:**
-    {context}
-
+    Improve the following code:
+    
     **Original Code:**
     {code}
 
-    **Enhanced Code with better efficiency, security, and structure:**
+    **Enhancements Required:**
+    - Optimize performance and efficiency.
+    - Improve security, error handling, and best practices.
+    - Add better structure and documentation.
+    - Ensure clean, readable formatting.
+
+    **Enhanced Code:**
     """
 )
 
-modify_template = PromptTemplate(
-    input_variables=["code", "modification", "context"],
+modify_prompt = PromptTemplate(
+    input_variables=["code", "modification"],
     template="""
-    Modify the code as per user requirements while maintaining context.
-
-    **Previous Conversation Context:**
-    {context}
+    Modify the following code based on user requests:
 
     **Original Code:**
     {code}
 
-    **User Requested Modification:**
+    **Modification Request:**
     {modification}
 
     **Modified Code:**
@@ -89,73 +88,65 @@ modify_template = PromptTemplate(
 )
 
 # ------------------------------- STREAMLIT UI -------------------------------
-st.title("🧠💻 Context-Aware Intelligent Coding Agent")
-st.markdown("### Generate, Enhance, and Modify Code While Retaining Context!")
+st.title("🧠💻 Intelligent Coding Agent")
+st.markdown("### Generate, Improve, and Modify Code in Any Programming Language!")
 
-languages = ["Python", "JavaScript", "PHP", "C#", "HTML/CSS/JS"]
+languages = [
+    "Python", "JavaScript", "TypeScript", "PHP", "C#", "Java", "C++", "Go", "Rust", "Swift", 
+    "Kotlin", "R", "SQL", "Bash", "HTML/CSS/JS"
+]
 language = st.selectbox("Select a programming language:", languages)
-description = st.text_area("Describe the code you need:", placeholder="E.g., Build a login system in PHP")
+description = st.text_area("What do you need?", placeholder="E.g., Build an authentication system in Flask")
 
 # ------------------------------- SESSION STATE -------------------------------
 if "generated_code" not in st.session_state:
     st.session_state.generated_code = ""
 
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
-
 # ------------------------------- ASYNC FUNCTIONS -------------------------------
 async def generate_code():
-    """Generates code while retaining context."""
+    """Generates optimized code while keeping a conversational flow."""
     if not description.strip():
-        st.warning("Please enter a code description!")
+        st.warning("Please enter a description!")
         return
     
-    st.info("⏳ Generating your code...")
-
-    # Retrieve past interactions for context
-    previous_context = "\n".join(st.session_state.chat_history)
+    st.info("🧑‍💻 Thinking and generating your code...")
 
     # Format prompt
-    prompt = prompt_template.format(language=language, description=description, context=previous_context)
+    prompt = code_prompt.format(language=language, description=description)
     
     # Generate code asynchronously
     response = await asyncio.to_thread(llm.invoke, [HumanMessage(content=prompt)])
 
-    # Store generated code and update chat history
+    # Store generated code and maintain conversation flow
     st.session_state.generated_code = response.content
-    st.session_state.chat_history.append(f"User asked: {description}\nAI Response:\n{response.content}")
 
-    # Display code
-    st.subheader("Generated Code:")
+    # Display code in a properly formatted block
+    st.subheader("Here’s your generated code:")
     st.code(st.session_state.generated_code, language=language.lower())
 
-    # Save to vectorstore for future reference
+    # Save to vectorstore for reference
     vectorstore.add_texts([st.session_state.generated_code])
 
 async def enhance_code():
-    """Enhances the generated code while considering past context."""
+    """Enhances existing code with better efficiency, structure, and readability."""
     if not st.session_state.generated_code:
         st.warning("Generate code first before enhancing!")
         return
 
-    st.info("⏳ Enhancing the code for better performance...")
-
-    previous_context = "\n".join(st.session_state.chat_history)
+    st.info("🔄 Improving your code...")
 
-    prompt = enhance_template.format(code=st.session_state.generated_code, context=previous_context)
+    prompt = enhance_prompt.format(code=st.session_state.generated_code)
     response = await asyncio.to_thread(llm.invoke, [HumanMessage(content=prompt)])
 
     st.session_state.generated_code = response.content
-    st.session_state.chat_history.append(f"AI Enhanced Code:\n{response.content}")
-
-    st.subheader("Enhanced Code:")
+    st.subheader("Here’s the improved version:")
     st.code(st.session_state.generated_code, language=language.lower())
 
 async def modify_code():
-    """Modifies existing code based on user input while remembering context."""
-    modification = st.text_area("Specify modifications (optional):", placeholder="E.g., Add validation for login form")
+    """Modifies code based on user feedback."""
+    modification = st.text_area("What changes do you want?", placeholder="E.g., Add input validation")
 
-    if st.button("Apply Modifications"):
+    if st.button("Apply Changes"):
         if not st.session_state.generated_code:
             st.warning("Generate code first before modifying!")
             return
@@ -163,17 +154,13 @@ async def modify_code():
             st.warning("Please enter modification details!")
             return
 
-        st.info("⏳ Modifying the code as per request...")
+        st.info("🔄 Making modifications...")
 
-        previous_context = "\n".join(st.session_state.chat_history)
-
-        prompt = modify_template.format(code=st.session_state.generated_code, modification=modification, context=previous_context)
+        prompt = modify_prompt.format(code=st.session_state.generated_code, modification=modification)
         response = await asyncio.to_thread(llm.invoke, [HumanMessage(content=prompt)])
 
         st.session_state.generated_code = response.content
-        st.session_state.chat_history.append(f"User Modification Request: {modification}\nAI Modified Code:\n{response.content}")
-
-        st.subheader("Modified Code:")
+        st.subheader("Here’s the modified code:")
         st.code(st.session_state.generated_code, language=language.lower())
 
 # ------------------------------- BUTTONS -------------------------------
@@ -194,9 +181,5 @@ with col3:
 if st.session_state.generated_code:
     st.download_button("Download Code", st.session_state.generated_code, "generated_code.txt", "text/plain")
 
-# ------------------------------- DISPLAY MEMORY -------------------------------
-st.subheader("Chat History & Context")
-st.text_area("Conversation Memory", value="\n".join(st.session_state.chat_history), height=200, disabled=True)
-
 st.markdown("---")
-st.markdown("🔹 **Built with LangChain (Optimized), OpenAI GPT, and ChromaDB** 🔹")
+st.markdown("🔹 **Powered by OpenAI GPT-4, LangChain, and ChromaDB** 🔹")