Update src/streamlit_app.py

src/streamlit_app.py

@@ -68,38 +68,7 @@ def create_download_link(file_path: str, label: str = None) -> str:
         return f"Error creating download link: {e}"
 
 
-def show_file(file_path: str):
-    """Show file inline + download link."""
-    if not os.path.exists(file_path):
-        return
-    ext = os.path.splitext(file_path)[1].lower()
-    if ext in [".png", ".jpg", ".jpeg"]:
-        st.image(file_path, caption=os.path.basename(file_path))
-    elif ext in [".txt", ".py", ".java", ".cpp", ".md"]:
-        with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
-            st.code(f.read(), language=ext.lstrip("."))
-    st.markdown(create_download_link(file_path), unsafe_allow_html=True)
-
-
-def render_tool_message(tool_message: ToolMessage):
-    """Render tool execution based on tool name instead of message content."""
-    file_related_keywords = ["read", "write", "file", "save", "export", "create"]
-    tool_name = getattr(tool_message, "name", "").lower()
-    st.info(f"🧰 Using tool :-  {tool_name or ''}")
-
-        # Check if this is a file-related tool
-    if any(k in tool_name for k in file_related_keywords):
-        # Find all files in TEMP_DIR (freshly modified ones)
-        created_files = sorted(
-            [os.path.join(TEMP_DIR, f) for f in os.listdir(TEMP_DIR)],
-            key=lambda x: os.path.getmtime(x),
-            reverse=True,
-        )
-        if created_files:
-            st.success("📄 File(s) created by tool:")
-            for file_path in created_files[:3]:
-                if("chatbot" not in file_path):
-                    show_file(file_path)
+
 
 
 def loadchats():
@@ -143,7 +112,11 @@ if "current_chat_id" in st.session_state:
                 if isinstance(message, AIMessage):
                     full_response += message.content or ""
                 elif isinstance(message, ToolMessage):
-                    render_tool_message(message)
+                    st.info("Using Appropriate tool")
+                    tool_name = message.tool_calls[0]['name'] 
+                    if(tool_name == "read_file" or tool_name == "write_file"):
+                        create_download_link(message.content) 
+                    
 
                 response_placeholder.markdown(full_response + "|")
         st.rerun()