Update app.py

app.py

@@ -1,9 +1,3 @@
-# .env file (create this in your project root)
-# -----------------------------------------
-# Your Gemini API key and model configuration
-
-
-# app.py
 import os
 from dotenv import load_dotenv
 import streamlit as st
@@ -11,28 +5,26 @@ from crewai import Agent, Task, Crew, Process
 from langchain_openai import ChatOpenAI
 
 # === Load environment variables ===
-load_dotenv()  # reads the .env file and loads variables into the environment
+load_dotenv()  # loads API_KEY and MODEL_NAME from .env
 
 # === API Key & Model Config ===
 API_KEY = os.getenv("API_KEY")
 MODEL_NAME = os.getenv("MODEL_NAME", "gemini-1.5")
-
-# Ensure the key is available for downstream libraries
-os.environ["OPENAI_API_KEY"] = API_KEY
+os.environ["OPENAI_API_KEY"] = API_KEY  # needed by langchain_openai
 
 # === Streamlit UI Config ===
 st.set_page_config(
-    page_title="👨‍💻 Java Coding Assistant",
-    page_icon="☕", 
+    page_title="☕ Java Coding Assistant",
+    page_icon="☕",
     layout="wide"
 )
 
-# === Session State ===
+# Initialize chat history
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# === Java Coding Assistant Function ===
-def java_coding_assistant(query, temperature=0.7):
+# Java Coding Assistant function
+def java_coding_assistant(query: str, temperature: float = 0.7) -> str:
     try:
         llm = ChatOpenAI(
             model_name=MODEL_NAME,
@@ -40,11 +32,11 @@ def java_coding_assistant(query, temperature=0.7):
             api_key=API_KEY
         )
 
-        # Main Java Expert Agent
         java_expert = Agent(
             role="Senior Java Full Stack Developer",
             goal="Provide expert-level Java solutions with clear explanations and best practices",
-            backstory="""You are a Java expert with 10+ years of experience in:
+            backstory="""
+You are a Java expert with 10+ years of experience in:
 - Core Java (Collections, Multithreading, JDBC)
 - Backend Development (Spring Boot, Hibernate, Microservices)
 - Android Development (Java/Kotlin, Android SDK)
@@ -53,28 +45,32 @@ You provide:
 1. Complete code solutions with syntax highlighting
 2. Detailed explanations of concepts
 3. Best practices and optimization tips
-4. Alternative approaches when applicable""",
+4. Alternative approaches when applicable
+""",
             llm=llm,
             verbose=True
         )
 
-        # Task for handling the query
         task = Task(
-            description=f"""Provide a comprehensive solution for this Java-related query:
+            description=f"""
+Provide a comprehensive solution for this Java-related query:
 {query}
 Include:
 1. Problem analysis
 2. Complete Java code solution with proper syntax
 3. Explanation of key concepts
 4. Best practices
-5. Potential edge cases""",
+5. Potential edge cases
+""",
             agent=java_expert,
-            expected_output="""A well-structured response containing:
+            expected_output="""
+A well-structured response containing:
 - Problem breakdown
 - Complete Java code with syntax highlighting
 - Detailed explanation
 - Best practices
-- Runtime complexity analysis (if applicable)"""
+- Runtime complexity analysis (if applicable)
+"""
         )
 
         crew = Crew(
@@ -85,29 +81,31 @@ Include:
         )
 
         result = crew.kickoff()
-        # Format code blocks if present
+
+        # Replace java fences for Streamlit
         if "```java" in result:
-            result = result.replace("```java", "```python")  # For Streamlit syntax highlighting
+            result = result.replace("```java", "```python")
         return result
 
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        return f"❌ Error: {e}"
 
-# === Streamlit UI ===
+# === Streamlit interface ===
 st.title("☕ Java Coding Assistant")
 st.caption("Ask me anything about Java backend, Android development, or coding problems")
 
-# Display chat history
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
+# Render chat history
+for msg in st.session_state.messages:
+    with st.chat_message(msg["role"]):
+        st.markdown(msg["content"])
 
 # Chat input
-if prompt := st.chat_input("Ask your Java question (e.g., 'How to implement REST API in Spring Boot?'):"):
-    st.session_state.messages.append({"role": "user", "content": prompt})
+if user_input := st.chat_input("Ask your Java question (e.g., 'How to implement REST API in Spring Boot?'):"):
+    st.session_state.messages.append({"role": "user", "content": user_input})
     with st.chat_message("user"):
-        st.markdown(prompt)
+        st.markdown(user_input)
     with st.chat_message("assistant"):
-        response = java_coding_assistant(prompt)
-        st.markdown(response, unsafe_allow_html=True)
-    st.session_state.messages.append({"role": "assistant", "content": response})
+        answer = java_coding_assistant(user_input)
+        st.markdown(answer, unsafe_allow_html=True)
+    st.session_state.messages.append({"role": "assistant", "content": answer})
+```