Update app.py

app.py

@@ -28,7 +28,6 @@ class State(TypedDict):
  
 # LLM
 llm = ChatGroq(model="deepseek-r1-distill-llama-70b")
-# llm_with_tools = llm.bind_tools(tools=tools)
  
 # Memory
 memory = MemorySaver()
@@ -41,11 +40,7 @@ def ai_assistance(state: State):
 # Build LangGraph
 builder = StateGraph(State)
 builder.add_node("AI_Assistance", ai_assistance)
-# builder.add_node("tools", ToolNode(tools))
- 
 builder.add_edge(START, "AI_Assistance")
-# builder.add_conditional_edges("AI_Assistance", tools_condition)
-# builder.add_edge("tools", "AI_Assistance")
 builder.add_edge("AI_Assistance", END)
 graph = builder.compile(checkpointer=memory)
  
@@ -100,8 +95,6 @@ if "thread_id" not in st.session_state:
     st.session_state.thread_id = "1"
 if "chat_history" not in st.session_state:
     st.session_state.chat_history = []
-if "show_upload" not in st.session_state:
-    st.session_state.show_upload = False
 if "selected_tools" not in st.session_state:
     st.session_state.selected_tools = ["Arxiv", "Wikipedia", "Tavily"]
  
@@ -113,32 +106,8 @@ for msg in st.session_state.chat_history:
 # Custom Chat Input Box UI
 with st.container():
     with st.form("chat_form", clear_on_submit=True):
-        st.markdown('<div id="floating-container">', unsafe_allow_html=True)
- 
-        col1, col2 = st.columns([1, 17])
-        with col1:
-            toggle = st.form_submit_button("➕", use_container_width=True)
-        with col2:
-            user_input = st.text_input("Ask me", label_visibility="collapsed",placeholder="Ask me Anything")
- 
-        st.markdown('</div>', unsafe_allow_html=True)
- 
-        # Floating Popup for Uploads and Tools
-        if toggle:
-            st.session_state.show_upload = not st.session_state.show_upload
- 
-        if st.session_state.show_upload:
-            with st.expander("📎 Upload Files or Images", expanded=False,width=300):
-                st.file_uploader("Upload:", type=["txt", "pdf", "docx"], key="text_file")
-                st.file_uploader("Upload image:", type=["jpg", "jpeg", "png"], key="img_file")
- 
-            # st.session_state.selected_tools = st.multiselect(
-            #     "🛠 Select Tools:",
-            #     ["Arxiv", "Wikipedia", "Tavily"],
-            #     default=st.session_state.selected_tools,
-            #     width= 300
-            # )
- 
+        st.markdown('<div id="floating-container">', unsafe_allow_html=True)        
+        st.markdown('</div>', unsafe_allow_html=True) 
         submitted = st.form_submit_button(label = "Send")
         if submitted and user_input:
             st.session_state.chat_history.append(HumanMessage(content=user_input))