Update app.py

app.py

@@ -2,15 +2,17 @@ import os
 from dotenv import load_dotenv
 import streamlit as st
 from crewai import Agent, Task, Crew, Process
-from langchain_openai import ChatOpenAI
+import google.generativeai as genai
 
 # === Load environment variables ===
-load_dotenv()  # loads API_KEY and MODEL_NAME from .env
+load_dotenv()  # loads API_KEY from .env
 
-# === API Key & Model Config ===
-API_KEY = os.getenv("API_KEY")
-MODEL_NAME = os.getenv("MODEL_NAME", "gemini-1.5")
-os.environ["OPENAI_API_KEY"] = API_KEY  # needed by langchain_openai
+# === API Key Config ===
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+os.environ["GEMINI_API_KEY"] = GEMINI_API_KEY  # Set for potential other uses
+
+# Configure Gemini
+genai.configure(api_key=GEMINI_API_KEY)
 
 # === Streamlit UI Config ===
 st.set_page_config(
@@ -26,63 +28,37 @@ if "messages" not in st.session_state:
 # Java Coding Assistant function
 def java_coding_assistant(query: str, temperature: float = 0.7) -> str:
     try:
-        llm = ChatOpenAI(
-            model_name=MODEL_NAME,
-            temperature=temperature,
-            api_key=API_KEY
-        )
-
-        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:
-- Core Java (Collections, Multithreading, JDBC)
-- Backend Development (Spring Boot, Hibernate, Microservices)
-- Android Development (Java/Kotlin, Android SDK)
-- System Design and Architecture
-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
-""",
-            llm=llm,
-            verbose=True
-        )
-
-        task = Task(
-            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
-""",
-            agent=java_expert,
-            expected_output="""
-A well-structured response containing:
-- Problem breakdown
-- Complete Java code with syntax highlighting
-- Detailed explanation
-- Best practices
-- Runtime complexity analysis (if applicable)
-"""
-        )
-
-        crew = Crew(
-            agents=[java_expert],
-            tasks=[task],
-            process=Process.sequential,
-            verbose=True
-        )
-
-        result = crew.kickoff()
-
-        # Replace java fences for Streamlit
+        # Initialize Gemini model
+        model = genai.GenerativeModel('gemini-1.5-pro')
+        
+        # Create prompt with the query
+        prompt = f"""
+        You are a Senior Java Full Stack Developer with 10+ years of experience.
+        Provide a comprehensive solution for this Java-related query:
+        {query}
+        
+        Your response should include:
+        1. Problem analysis
+        2. Complete Java code solution with proper syntax
+        3. Explanation of key concepts
+        4. Best practices
+        5. Potential edge cases
+        
+        Format your response with:
+        - Problem breakdown
+        - Complete Java code with syntax highlighting (use ```java markers)
+        - Detailed explanation
+        - Best practices
+        - Runtime complexity analysis (if applicable)
+        """
+        
+        # Generate response
+        response = model.generate_content(prompt)
+        
+        # Get the text from the response
+        result = response.text
+        
+        # Replace java fences for Streamlit if needed
         if "```java" in result:
             result = result.replace("```java", "```python")
         return result
@@ -107,4 +83,4 @@ if user_input := st.chat_input("Ask your Java question (e.g., 'How to implement
     with st.chat_message("assistant"):
         answer = java_coding_assistant(user_input)
         st.markdown(answer, unsafe_allow_html=True)
-    st.session_state.messages.append({"role": "assistant", "content": answer})
+    st.session_state.messages.append({"role": "assistant", "content": answer})