Update app.py

app.py

@@ -1,47 +1,77 @@
 import streamlit as st
 import os
 import asyncio
+import time
 from langchain_openai import ChatOpenAI, OpenAIEmbeddings
 from langchain_chroma import Chroma
 from langchain.prompts import PromptTemplate
-from langchain.memory import ConversationBufferMemory
 from langchain.schema import AIMessage, HumanMessage
-import re
+
 # ------------------------------- CONFIG -------------------------------
 st.set_page_config(page_title="Intelligent Coding Agent", layout="wide")
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")  # Ensure API key is set
+
+# Inject custom CSS for full-width layout and animations
+st.markdown("""
+    <style>
+        .appview-container .main .block-container {
+            max-width: 100%;
+            padding: 0 2rem;
+        }
+        .stTextArea, .stTextInput, .stButton {
+            width: 100% !important;
+        }
+        .stCodeBlock {
+            white-space: pre-wrap !important;
+            width: 100%;
+        }
+        .branding {
+            text-align: center;
+            font-weight: bold;
+            font-size: 14px;
+            color: #888;
+            padding: 10px;
+        }
+        .loader {
+            text-align: center;
+            font-size: 18px;
+            font-weight: bold;
+            animation: fadeIn 1s infinite alternate;
+        }
+        @keyframes fadeIn {
+            from {opacity: 0.3;}
+            to {opacity: 1;}
+        }
+    </style>
+""", unsafe_allow_html=True)
 
 # ------------------------------- CACHING -------------------------------
 @st.cache_resource
 def get_openai_model():
-    """Load OpenAI model once (cached) for efficiency."""
-    return ChatOpenAI(model_name="gpt-4-turbo", temperature=0.7)
+    """Load OpenAI GPT-3.5-Turbo model once (cached) for efficiency."""
+    return ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.7)
 
 @st.cache_resource
 def get_vectorstore():
-    """Load ChromaDB once (cached) for faster retrieval."""
+    """Load ChromaDB once (cached) for fast retrieval."""
     return Chroma(
         embedding_function=OpenAIEmbeddings(),
         persist_directory="/home/user/chroma_db"
     )
 
-@st.cache_resource
-def get_memory():
-    """Initialize memory for conversational interactions."""
-    return ConversationBufferMemory(memory_key="chat_history", return_messages=True)
-
-# Initialize models and memory
+# Initialize models
 llm = get_openai_model()
 vectorstore = get_vectorstore()
-memory = get_memory()
 
-# ------------------------------- STRONGER PROMPT TEMPLATES -------------------------------
+# ------------------------------- PROMPT TEMPLATE -------------------------------
 code_prompt = PromptTemplate(
     input_variables=["language", "description"],
     template="""
-    You are an **expert AI coding assistant** specializing in **ALL programming languages**.
+    You are an **expert AI coding assistant** specializing in **ALL programming languages**, including:
+    - Python, JavaScript, TypeScript, PHP, C#, Java, C++, Go, Rust, Swift, Kotlin, R, SQL, Bash
+    - .NET, .NET Core, ASP.NET, Node.js, Django, Flask, Spring Boot, Ruby on Rails
+    - Web technologies: HTML, CSS, WebAssembly, WebSockets, Web3.js, GraphQL, REST APIs
 
-    **Task:** Generate an optimized, structured, and functional code snippet.
+    **Task:** Generate a well-structured, optimized, and fully functional code snippet.
 
     **Language:** {language}  
     **User Request:** {description}  
@@ -49,69 +79,41 @@ code_prompt = PromptTemplate(
     - 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_prompt = PromptTemplate(
-    input_variables=["code"],
-    template="""
-    Improve the following code:
-    
-    **Original Code:**
-    {code}
-
-    **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_prompt = PromptTemplate(
-    input_variables=["code", "modification"],
-    template="""
-    Modify the following code based on user requests:
-
-    **Original Code:**
-    {code}
-
-    **Modification Request:**
-    {modification}
-
-    **Modified Code:**
+    - Ensure security and performance optimizations.
     """
 )
 
 # ------------------------------- STREAMLIT UI -------------------------------
 st.title("🧠💻 Intelligent Coding Agent")
-st.markdown("### Generate, Improve, and Modify Code in Any Programming Language!")
+st.markdown("### Generate, Enhance, and Modify Code Seamlessly!")
 
 languages = [
     "Python", "JavaScript", "TypeScript", "PHP", "C#", "Java", "C++", "Go", "Rust", "Swift", 
-    "Kotlin", "R", "SQL", "Bash", "HTML/CSS/JS"
+    "Kotlin", "R", "SQL", "Bash", "HTML/CSS/JS", ".NET", ".NET Core"
 ]
 language = st.selectbox("Select a programming language:", languages)
-description = st.text_area("What do you need?", placeholder="E.g., Build an authentication system in Flask")
+description = st.text_area("What do you need?", placeholder="E.g., Build an authentication system in Flask", height=150)
 
 # ------------------------------- SESSION STATE -------------------------------
 if "generated_code" not in st.session_state:
     st.session_state.generated_code = ""
 
-# ------------------------------- ASYNC FUNCTIONS -------------------------------
-
+# ------------------------------- ANIMATED LOADER -------------------------------
+def show_loader():
+    """Shows an animated loading effect."""
+    loader_text = "⏳ Generating your code..."
+    for _ in range(3):
+        st.markdown(f"<div class='loader'>{loader_text}</div>", unsafe_allow_html=True)
+        time.sleep(0.6)
 
+# ------------------------------- ASYNC FUNCTIONS -------------------------------
 async def generate_code():
-    """Generates optimized code while ensuring correct formatting."""
+    """Generates optimized code while ensuring proper formatting."""
     if not description.strip():
         st.warning("Please enter a description!")
         return
     
-    st.info("🧑‍💻 Thinking and generating your code...")
+    show_loader()
 
     # Format prompt
     prompt = code_prompt.format(language=language, description=description)
@@ -119,30 +121,25 @@ async def generate_code():
     # Generate code asynchronously
     response = await asyncio.to_thread(llm.invoke, [HumanMessage(content=prompt)])
 
-    # Extract only the code part (removes ``` and language tag)
-    extracted_code = re.sub(r"```[\w+]*\n", "", response.content)  # Remove triple backticks & language tag
-    extracted_code = re.sub(r"```$", "", extracted_code.strip())  # Remove trailing backticks
-
-    # Store formatted code in session state
-    st.session_state.generated_code = extracted_code
+    # Store generated code in session state
+    st.session_state.generated_code = response.content
 
-    # Display code in a properly formatted block
+    # Display properly formatted code block
     st.subheader("Here’s your generated code:")
     st.code(st.session_state.generated_code, language=language.lower())
 
     # Save to vectorstore for reference
     vectorstore.add_texts([st.session_state.generated_code])
 
-
 async def enhance_code():
-    """Enhances existing code with better efficiency, structure, and readability."""
+    """Enhances existing code with better efficiency and readability."""
     if not st.session_state.generated_code:
         st.warning("Generate code first before enhancing!")
         return
 
     st.info("🔄 Improving your code...")
 
-    prompt = enhance_prompt.format(code=st.session_state.generated_code)
+    prompt = f"Improve the following code for better efficiency and readability:\n\n{st.session_state.generated_code}"
     response = await asyncio.to_thread(llm.invoke, [HumanMessage(content=prompt)])
 
     st.session_state.generated_code = response.content
@@ -163,7 +160,7 @@ async def modify_code():
 
         st.info("🔄 Making modifications...")
 
-        prompt = modify_prompt.format(code=st.session_state.generated_code, modification=modification)
+        prompt = f"Modify the following code based on the request '{modification}':\n\n{st.session_state.generated_code}"
         response = await asyncio.to_thread(llm.invoke, [HumanMessage(content=prompt)])
 
         st.session_state.generated_code = response.content
@@ -171,22 +168,22 @@ async def modify_code():
         st.code(st.session_state.generated_code, language=language.lower())
 
 # ------------------------------- BUTTONS -------------------------------
-col1, col2, col3 = st.columns(3)
+col1, col2 = st.columns([2, 1])
 
 with col1:
-    if st.button("Generate Code"):
+    if st.button("Generate Code", use_container_width=True):
         asyncio.run(generate_code())
 
 with col2:
-    if st.button("Enhance Code"):
+    if st.button("Enhance Code", use_container_width=True):
         asyncio.run(enhance_code())
 
-with col3:
-    modify_code()
-
 # ------------------------------- DOWNLOAD OPTION -------------------------------
 if st.session_state.generated_code:
-    st.download_button("Download Code", st.session_state.generated_code, "generated_code.txt", "text/plain")
+    st.download_button("Download Code", st.session_state.generated_code, "generated_code.txt", "text/plain", use_container_width=True)
+
+# ------------------------------- BRANDING -------------------------------
+st.markdown('<p class="branding">🚀 Powered by WaysAhead Global</p>', unsafe_allow_html=True)
 
 st.markdown("---")
-st.markdown("🔹 **Powered by OpenAI GPT-4, LangChain, and ChromaDB** 🔹")
+st.markdown("🔹 **Optimized with OpenAI GPT-3.5, LangChain, and ChromaDB** 🔹")