Update app.py

app.py

@@ -31,7 +31,7 @@ class State(TypedDict):
     explanation:str
  
 # LLM
-llm = ChatGroq(model="deepseek-r1-distill-llama-70b")
+# llm = ChatGroq(model="deepseek-r1-distill-llama-70b")
 code_generator = Together(
     model="meta-llama/Llama-3.3-70B-Instruct-Turbo-Free",  
     temperature=0.2,
@@ -44,7 +44,7 @@ memory = MemorySaver()
  
 # Define node
 def ai_assistance(state: State):
-    result = llm.invoke(state["messages"])
+    result =code_generator.invoke(state["messages"])
     return {"messages": state['messages']+[result]}
 import time
 def generate_code(state: State):
@@ -132,17 +132,12 @@ def explain_code(state):
     Explain the LangGraph flow, code architecture, and provide detailed instructions for visualization."""
     
     full_prompt = system_prompt + prompt
-    explanation = llm.invoke(full_prompt)
+    explanation = code_generator.invoke(full_prompt)
     
     return {"input": user_prompt, "code": code, "explanation": explanation}
 
 
 
-# def router(state: State):
-#     last_msg = state["messages"][-1].content.lower()
-#     if any(keyword in last_msg for keyword in ["generate", "code", "script", "langgraph", "flow"]):
-#         return "Generate_Code"
-#     return "AI_Assistance"
 
 def router(state):
     user_input = state["input"].lower()
@@ -164,7 +159,7 @@ graph = builder.compile(checkpointer=memory)
 
  
 # Streamlit UI setup
-st.set_page_config(page_title="LangGraph Chatbot", layout="wide")
+st.set_page_config(page_title="MitraVerse", layout="wide")
  
 st.markdown("""
 <style>
@@ -207,7 +202,7 @@ st.markdown("""
 </style>
 """, unsafe_allow_html=True)
  
-st.title("🧠 LangGraph Chatbot (Groq + Tools)")
+st.title("🧠MitraVerse")
  
 # Initialize session
 if "thread_id" not in st.session_state:
@@ -248,36 +243,5 @@ with st.container():
                 st.session_state.chat_history.append(result["messages"][-1])
 
 
-            # if result.get("code"):
-            #     st.session_state.chat_history.append(AIMessage(content=result["code"]))
-            
-            # if result.get("explanation"):
-            #     st.session_state.chat_history.append(AIMessage(content=result["explanation"]))
-            
-            # if result.get("messages"):
-            #     st.session_state.chat_history.append(result["messages"][-1])
-
-        
-            # If it’s a code response, append it as-is
-            # if result.get("code"):
-            #     st.session_state.chat_history.append(AIMessage(content=result["code"]))
-            # else:
-            #     st.session_state.chat_history.append(result["messages"][-1])
-            # if result.get("code"):
-            #     st.session_state.chat_history.append(AIMessage(content=result["code"]))
-            # elif result.get("explanation"):
-            #     st.session_state.chat_history.append(AIMessage(content=result["explanation"]))
-            # elif result.get("messages"):
-            #     st.session_state.chat_history.append(result["messages"][-1])
-
-            # if result.get("code"):
-            #     st.session_state.chat_history.append(AIMessage(content=result["code"]))
-            
-            # if result.get("explanation"):
-            #     st.session_state.chat_history.append(AIMessage(content="**🔍 Code Explanation:**\n\n" + result["explanation"]))
-            
-            # if result.get("messages"):
-            #     st.session_state.chat_history.append(result["messages"][-1])
-
 
             st.rerun()